XYZ plot do not fail if an exception occurs (401ba1b8) · Commits · github_fork / Stable Diffusion Webui

scripts/xyz_grid.py

+7 −2

Original line number	Diff line number	Diff line
		@@ -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)

		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