From 94db86402c2ab921db50372a3b21b43e02d27a04 Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 9 Aug 2023 15:09:40 -0700 Subject: [PATCH] i --- sgm/modules/diffusionmodules/util.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)