Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin
2023-08-09 15:07:09 -07:00
parent c56e2c9fd1
commit c104f30137
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
entry.py
+1 -1
View File
@@ -151,6 +151,6 @@ with torch.no_grad():
model.first_stage_model.cpu() model.first_stage_model.cpu()
import cv2 import cv2
samples = einops.rearrange(samples, 'b c h w -> b h w c')[0] * 255.0 samples = einops.rearrange(samples, 'b c h w -> b h w c')[0, :, :, ::-1] * 255.0
samples = samples.cpu().numpy().clip(0, 255).astype(np.uint8) samples = samples.cpu().numpy().clip(0, 255).astype(np.uint8)
cv2.imwrite('img.png', samples) cv2.imwrite('img.png', samples)
img.png
BIN
View File
Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 4.7 KiB

After

Width:  |  Height:  |  Size: 1.5 MiB

sgm/modules/diffusionmodules/util.py
+3 -1
View File
@@ -272,7 +272,9 @@ class SiLU(nn.Module):
class GroupNorm32(nn.GroupNorm): class GroupNorm32(nn.GroupNorm):
def forward(self, x): def forward(self, x):
return super().forward(x) self.weight.to(torch.float32)
self.bias.to(torch.float32)
return super().forward(x.float()).type(x.dtype)
def conv_nd(dims, *args, **kwargs): def conv_nd(dims, *args, **kwargs):