mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
i
This commit is contained in:
@@ -272,8 +272,7 @@ class SiLU(nn.Module):
|
|||||||
|
|
||||||
class GroupNorm32(nn.GroupNorm):
|
class GroupNorm32(nn.GroupNorm):
|
||||||
def forward(self, x):
|
def forward(self, x):
|
||||||
self.weight = self.weight.float()
|
self.to(torch.float32)
|
||||||
self.bias = self.bias.float()
|
|
||||||
return super().forward(x.float()).type(x.dtype)
|
return super().forward(x.float()).type(x.dtype)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user