From b3fbfd72a8f647161685b28d20b4b61519d8a643 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 15 Apr 2024 21:49:51 +0200 Subject: Update transformer model --- text_recognizer/network/transformer/norm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/network/transformer/norm.py') 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): -- cgit v1.2.3-70-g09d2