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-10 13:16:25 -07:00
parent 6d376f2fb0
commit a169dd2c77
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/core.py
+1 -1
View File
@@ -66,7 +66,7 @@ def get_previewer(device, latent_format):
def preview_function(x0, step, total_steps): def preview_function(x0, step, total_steps):
global cv2_is_top global cv2_is_top
with torch.no_grad(): with torch.no_grad():
x_sample = taesd.decoder(x0).detach() * 255.0 x_sample = taesd.decoder(torch.nn.functional.avg_pool2d(x0, kernel_size=(2, 2))).detach() * 255.0
x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c') x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c')
x_sample = x_sample.cpu().numpy()[..., ::-1].copy().clip(0, 255).astype(np.uint8) x_sample = x_sample.cpu().numpy()[..., ::-1].copy().clip(0, 255).astype(np.uint8)
for i, s in enumerate(x_sample): for i, s in enumerate(x_sample):