diff options
Diffstat (limited to 'text_recognizer/network/transformer/norm.py')
-rw-r--r-- | text_recognizer/network/transformer/norm.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/network/transformer/norm.py b/text_recognizer/network/transformer/norm.py index 2737754..9ba35a8 100644 --- a/text_recognizer/network/transformer/norm.py +++ b/text_recognizer/network/transformer/norm.py @@ -5,8 +5,8 @@ Copied from lucidrains: """ import torch -from torch import Tensor, nn import torch.nn.functional as F +from torch import Tensor, nn class RMSNorm(nn.Module): |