diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:49:51 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:49:51 +0200 |
commit | b3fbfd72a8f647161685b28d20b4b61519d8a643 (patch) | |
tree | a5cac4e15186396aae35231d6d6fe266691b0186 /text_recognizer/network/transformer/swiglu.py | |
parent | c7e5354ffa43eccfc4e411375ce2f531af7bbcff (diff) |
Update transformer model
Diffstat (limited to 'text_recognizer/network/transformer/swiglu.py')
-rw-r--r-- | text_recognizer/network/transformer/swiglu.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/network/transformer/swiglu.py b/text_recognizer/network/transformer/swiglu.py index e61662a..7bafd06 100644 --- a/text_recognizer/network/transformer/swiglu.py +++ b/text_recognizer/network/transformer/swiglu.py @@ -1,5 +1,5 @@ -from torch import nn import torch.nn.functional as F +from torch import nn class SwiGLU(nn.Module): |