fix for img2img color correction in a batch #3218 (ca5a9e79) · Commits · github_fork / Stable Diffusion Webui

modules/processing.py

+4 −2

Original line number	Diff line number	Diff line
		@@ -403,8 +403,6 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
		if (len(prompts) == 0):
		break

		#uc = p.sd_model.get_learned_conditioning(len(prompts) * [p.negative_prompt])
		#c = p.sd_model.get_learned_conditioning(prompts)
		with devices.autocast():
		uc = prompt_parser.get_learned_conditioning(shared.sd_model, len(prompts) * [p.negative_prompt], p.steps)
		c = prompt_parser.get_multicond_learned_conditioning(shared.sd_model, prompts, p.steps)
		@@ -716,6 +714,10 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):
		batch_images = np.expand_dims(imgs[0], axis=0).repeat(self.batch_size, axis=0)
		if self.overlay_images is not None:
		self.overlay_images = self.overlay_images * self.batch_size

		if self.color_corrections is not None and len(self.color_corrections) == 1:
		self.color_corrections = self.color_corrections * self.batch_size

		elif len(imgs) <= self.batch_size:
		self.batch_size = len(imgs)
		batch_images = np.array(imgs)