diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-01 06:22:53 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-01 06:22:53 +0000 |
| commit | 401ba1b879ecb36b556d677ba66881ee9cbe7fbf (patch) | |
| tree | c144804f2658dddbd0a864aa0926f30e8b17afc2 | |
| parent | 07be13caa357b14f6afa247566d53339522b8e66 (diff) | |
| download | stable-diffusion-webui-gfx803-401ba1b879ecb36b556d677ba66881ee9cbe7fbf.tar.gz stable-diffusion-webui-gfx803-401ba1b879ecb36b556d677ba66881ee9cbe7fbf.tar.bz2 stable-diffusion-webui-gfx803-401ba1b879ecb36b556d677ba66881ee9cbe7fbf.zip | |
XYZ plot do not fail if an exception occurs
| -rw-r--r-- | scripts/xyz_grid.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py index d3c78a2f..103bf104 100644 --- a/scripts/xyz_grid.py +++ b/scripts/xyz_grid.py @@ -11,7 +11,7 @@ import numpy as np import modules.scripts as scripts
import gradio as gr
-from modules import images, sd_samplers, processing, sd_models, sd_vae, sd_samplers_kdiffusion
+from modules import images, sd_samplers, processing, sd_models, sd_vae, sd_samplers_kdiffusion, errors
from modules.processing import process_images, Processed, StableDiffusionProcessingTxt2Img
from modules.shared import opts, state
import modules.shared as shared
@@ -651,7 +651,12 @@ class Script(scripts.Script): y_opt.apply(pc, y, ys)
z_opt.apply(pc, z, zs)
- res = process_images(pc)
+ try:
+ res = process_images(pc)
+ except Exception as e:
+ errors.display(e, "generating image for xyz plot")
+
+ res = Processed(p, [], p.seed, "")
# Sets subgrid infotexts
subgrid_index = 1 + iz
|