diff options
Diffstat (limited to 'text_recognizer/networks/transformer/ff.py')
| -rw-r--r-- | text_recognizer/networks/transformer/ff.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/text_recognizer/networks/transformer/ff.py b/text_recognizer/networks/transformer/ff.py index 4028ab3..3ccf5b5 100644 --- a/text_recognizer/networks/transformer/ff.py +++ b/text_recognizer/networks/transformer/ff.py @@ -5,9 +5,8 @@ Stolen from lucidrains: """ from typing import Optional -from torch import nn -from torch import Tensor import torch.nn.functional as F +from torch import Tensor, nn class GEGLU(nn.Module): |