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)