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):