diff --git a/sgm/modules/diffusionmodules/util.py b/sgm/modules/diffusionmodules/util.py index e4e4f14d..6c42a42e 100644 --- a/sgm/modules/diffusionmodules/util.py +++ b/sgm/modules/diffusionmodules/util.py @@ -272,8 +272,8 @@ class SiLU(nn.Module): class GroupNorm32(nn.GroupNorm): def forward(self, x): - self.weight.to(torch.float32) - self.bias.to(torch.float32) + self.weight = self.weight.float() + self.bias = self.bias.float() return super().forward(x.float()).type(x.dtype)