From a169dd2c7767603895c7c6c30ff44f72799c6126 Mon Sep 17 00:00:00 2001 From: lvmin Date: Thu, 10 Aug 2023 13:16:25 -0700 Subject: [PATCH] i --- modules/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/core.py b/modules/core.py index aba97861..7e8d27c1 100644 --- a/modules/core.py +++ b/modules/core.py @@ -66,7 +66,7 @@ def get_previewer(device, latent_format): def preview_function(x0, step, total_steps): global cv2_is_top 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 = x_sample.cpu().numpy()[..., ::-1].copy().clip(0, 255).astype(np.uint8) for i, s in enumerate(x_sample):