From e739e44ca06caace73e9f3dce00a24ef7a296f2f Mon Sep 17 00:00:00 2001 From: lvmin Date: Thu, 10 Aug 2023 21:10:05 -0700 Subject: [PATCH] i --- modules/core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/core.py b/modules/core.py index 99762469..f2154732 100644 --- a/modules/core.py +++ b/modules/core.py @@ -81,7 +81,10 @@ def get_previewer(device, latent_format): x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c') x_sample = x_sample.cpu().numpy().clip(0, 255).astype(np.uint8) for i, s in enumerate(x_sample): - show_preview(f'OpenCV Diffusion Preview {i}', s, title=f'Preview Image {i} [{step}/{total_steps}]') + if i > 0: + show_preview(f'OpenCV Diffusion Preview {i}', s, title=f'Preview Image {i} [{step}/{total_steps}]') + else: + show_preview(f'OpenCV Diffusion Preview {i}', s, title=f'Preview Image [{step}/{total_steps}]') taesd.preview = preview_function