Handle cleanup in case there's an exception thrown

        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

                    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 []

        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

                    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 []