From f9bc83733b5a3490b0be5e731f75bdfe5f276be4 Mon Sep 17 00:00:00 2001 From: Alireza Davoudi Date: Mon, 5 Sep 2022 21:00:09 +0430 Subject: [PATCH] Update model.py --- clip/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clip/model.py b/clip/model.py index 232b779..e71bbc2 100644 --- a/clip/model.py +++ b/clip/model.py @@ -159,7 +159,7 @@ class LayerNorm(nn.LayerNorm): def forward(self, x: torch.Tensor): orig_type = x.dtype - ret = super().forward(x.type(torch.float32)) + ret = super(LayerNorm).forward(x.type(torch.float32)) return ret.type(orig_type)