bugfix: save image for hires fix BEFORE upscaling latent space (f674c488) · Commits · github_fork / Stable Diffusion Webui

modules/processing.py

+4 −4

Original line number	Diff line number	Diff line
		@@ -665,6 +665,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
		images.save_image(image, self.outpath_samples, "", seeds[index], prompts[index], opts.samples_format, suffix="-before-highres-fix")

		if opts.use_scale_latent_for_hires_fix:
		for i in range(samples.shape[0]):
		save_intermediate(samples, i)

		samples = torch.nn.functional.interpolate(samples, size=(self.height // opt_f, self.width // opt_f), mode="bilinear")

		# Avoid making the inpainting conditioning unless necessary as
		@@ -673,9 +676,6 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
		image_conditioning = self.img2img_image_conditioning(decode_first_stage(self.sd_model, samples), samples)
		else:
		image_conditioning = self.txt2img_image_conditioning(samples)

		for i in range(samples.shape[0]):
		save_intermediate(samples, i)
		else:
		decoded_samples = decode_first_stage(self.sd_model, samples)
		lowres_samples = torch.clamp((decoded_samples + 1.0) / 2.0, min=0.0, max=1.0)