From b60c67ecd44c09257b6b79ffbf20cb6b7c2c3248 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 18 Jun 2026 08:49:46 +0200 Subject: [PATCH] fix preview mapping Signed-off-by: Vladimir Mandic --- data/previews.json | 1 + modules/progress.py | 11 +++++++---- scripts/pulid_ext.py | 2 +- 3 files changed, 9 insertions(+), 5 deletions(-) diff --git a/data/previews.json b/data/previews.json index b6a21e69e..08ab22a0e 100644 --- a/data/previews.json +++ b/data/previews.json @@ -1,5 +1,6 @@ { "sd-v21-512-ema": "models/Reference/stabilityai--stable-diffusion-2-1-base.jpg", + "stable-diffusion-v1-5--stable-diffusion-v1-5": "models/Reference/v1-5-pruned-fp16-emaonly.jpg", "stabilityai--stable-diffusion-xl-base-1.0": "models/Reference/stabilityai--stable-diffusion-xl-base-1.0.jpg", "stabilityai--stable-diffusion-3-medium-diffusers": "models/Reference/stabilityai--stable-diffusion-3.jpg", "stabilityai--stable-diffusion-3.5-medium": "models/Reference/stabilityai--stable-diffusion-3_5-medium.jpg", diff --git a/modules/progress.py b/modules/progress.py index 2fcd4dc22..5c2c3c7ec 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -97,10 +97,13 @@ def api_progress(req: ProgressRequest): if active and (req.id_live_preview != -1): have_image = shared.state.set_current_image() if have_image and shared.state.current_image is not None: - buffered = io.BytesIO() - shared.state.current_image.save(buffered, format='jpeg', quality=60) - b64 = base64.b64encode(buffered.getvalue()) - live_preview = f'data:image/jpeg;base64,{b64.decode("ascii")}' + try: + buffered = io.BytesIO() + shared.state.current_image.save(buffered, format='jpeg', quality=60) + b64 = base64.b64encode(buffered.getvalue()) + live_preview = f'data:image/jpeg;base64,{b64.decode("ascii")}' + except Exception: + live_preview = None else: live_preview = None diff --git a/scripts/pulid_ext.py b/scripts/pulid_ext.py index d9d312819..68880f06c 100644 --- a/scripts/pulid_ext.py +++ b/scripts/pulid_ext.py @@ -29,7 +29,7 @@ class PulIDScript(scripts_manager.Script): return True def dependencies(self): - from installer import installed, install, install_insightface + from installer import install, install_insightface install_insightface() install('torchdiffeq', quiet=True) install('torchsde==0.2.6', 'torchsde', quiet=True)