From 06a96f70abfca40fa51c73ce50a2db230238d904 Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 9 Aug 2023 15:11:01 -0700 Subject: [PATCH] i --- sgm/modules/diffusionmodules/util.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sgm/modules/diffusionmodules/util.py b/sgm/modules/diffusionmodules/util.py index 6c42a42e..6783480a 100644 --- 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)