Unverified Commit 45c0f58d authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #11923 from AnyISalIn/dev

[bug] If txt2img/img2img raises an exception, finally call state.end()
parents 075934a9 1fe2dcaa
Loading
Loading
Loading
Loading
+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 []