[bug] If txt2img/img2img raises an exception, finally call state.end() (1fe2dcaa) · Commits · github_fork / Stable Diffusion Webui

modules/api/api.py

+20 −16

Original line number	Diff line number	Diff line
		@@ -333,6 +333,7 @@ class Api:
		p.outpath_grids = opts.outdir_txt2img_grids
		p.outpath_samples = opts.outdir_txt2img_samples

		try:
		shared.state.begin(job="scripts_txt2img")
		if selectable_scripts is not None:
		p.script_args = script_args
		@@ -340,6 +341,7 @@ class Api:
		else:
		p.script_args = tuple(script_args) # Need to pass args as tuple here
		processed = process_images(p)
		finally:
		shared.state.end()

		b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []
		@@ -390,6 +392,7 @@ class Api:
		p.outpath_grids = opts.outdir_img2img_grids
		p.outpath_samples = opts.outdir_img2img_samples

		try:
		shared.state.begin(job="scripts_img2img")
		if selectable_scripts is not None:
		p.script_args = script_args
		@@ -397,6 +400,7 @@ class Api:
		else:
		p.script_args = tuple(script_args) # Need to pass args as tuple here
		processed = process_images(p)
		finally:
		shared.state.end()

		b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []