Handle cleanup in case there's an exception thrown

modules/api/api.py

@@ -323,6 +323,7 @@ class Api:
 
         with self.queue_lock:
             p = StableDiffusionProcessingTxt2Img(sd_model=shared.sd_model, **args)
+            try:
                 p.scripts = script_runner
                 p.outpath_grids = opts.outdir_txt2img_grids
                 p.outpath_samples = opts.outdir_txt2img_samples
@@ -335,6 +336,7 @@ class Api:
                     p.script_args = tuple(script_args) # Need to pass args as tuple here
                     processed = process_images(p)
                 shared.state.end()
+            finally:
                 p.close()
 
         b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []
@@ -380,6 +382,7 @@ class Api:
 
         with self.queue_lock:
             p = StableDiffusionProcessingImg2Img(sd_model=shared.sd_model, **args)
+            try:
                 p.init_images = [decode_base64_to_image(x) for x in init_images]
                 p.scripts = script_runner
                 p.outpath_grids = opts.outdir_img2img_grids
@@ -393,6 +396,8 @@ class Api:
                     p.script_args = tuple(script_args) # Need to pass args as tuple here
                     processed = process_images(p)
                 shared.state.end()
+
+            finally:
                 p.close()
 
         b64images = list(map(encode_pil_to_base64, processed.images)) if send_images else []