i

2026-08-16 13:13:16 +02:00 · 2023-08-09 14:27:40 -07:00
parent 8807b2dff8
commit 69f57c0454
3 changed files with 12 additions and 7 deletions
@@ -272,7 +272,7 @@ class SiLU(nn.Module):

 class GroupNorm32(nn.GroupNorm):
    def forward(self, x):
-        return super().forward(x.float()).type(x.dtype)
+        return super().forward(x)


 def conv_nd(dims, *args, **kwargs):