i

2023-08-09 15:11:01 -07:00 · 2023-08-09 15:11:01 -07:00 · 06a96f70ab
parent 94db86402c
commit 06a96f70ab
1 changed files with 1 additions and 2 deletions
--- a/sgm/modules/diffusionmodules/util.py
+++ b/sgm/modules/diffusionmodules/util.py
@ -272,8 +272,7 @@ class SiLU(nn.Module):

 class GroupNorm32(nn.GroupNorm):
    def forward(self, x):
-        self.weight = self.weight.float()
-        self.bias = self.bias.float()
+        self.to(torch.float32)
        return super().forward(x.float()).type(x.dtype)