Merge pull request #4142 from jn-jairo/processing-close (65522ff1) · Commits · github_fork / Stable Diffusion Webui

modules/img2img.py

+2 −0

Original line number	Diff line number	Diff line
		@@ -137,6 +137,8 @@ def img2img(mode: int, prompt: str, negative_prompt: str, prompt_style: str, pro
		if processed is None:
		processed = process_images(p)

		p.close()

		shared.total_tqdm.clear()

		generation_info_js = processed.js()

+4 −3

Original line number	Diff line number	Diff line
		@@ -202,6 +202,10 @@ class StableDiffusionProcessing():
		def sample(self, conditioning, unconditional_conditioning, seeds, subseeds, subseed_strength):
		raise NotImplementedError()

		def close(self):
		self.sd_model = None
		self.sampler = None


		class Processed:
		def __init__(self, p: StableDiffusionProcessing, images_list, seed=-1, info="", subseed=None, all_prompts=None, all_seeds=None, all_subseeds=None, index_of_first_image=0, infotexts=None):
		@@ -597,9 +601,6 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
		if p.scripts is not None:
		p.scripts.postprocess(p, res)

		p.sd_model = None
		p.sampler = None

		return res

+2 −0

Original line number	Diff line number	Diff line
		@@ -47,6 +47,8 @@ def txt2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2:
		if processed is None:
		processed = process_images(p)

		p.close()

		shared.total_tqdm.clear()

		generation_info_js = processed.js()