add is_hr_pass field for processing (7428fb51) · Commits · github_fork / Stable Diffusion Webui

modules/processing.py

+5 −0

Original line number	Original line	Diff line number	Diff line
	@@ -164,6 +164,7 @@ class StableDiffusionProcessing:
	self.all_seeds = None		self.all_seeds = None
	self.all_subseeds = None		self.all_subseeds = None
	self.iteration = 0		self.iteration = 0
			self.is_hr_pass = False


	@property		@property
	@@ -883,6 +884,8 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
	if not self.enable_hr:		if not self.enable_hr:
	return samples		return samples

			self.is_hr_pass = True

	target_width = self.hr_upscale_to_x		target_width = self.hr_upscale_to_x
	target_height = self.hr_upscale_to_y		target_height = self.hr_upscale_to_y

	@@ -952,6 +955,8 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):

	samples = self.sampler.sample_img2img(self, samples, noise, conditioning, unconditional_conditioning, steps=self.hr_second_pass_steps or self.steps, image_conditioning=image_conditioning)		samples = self.sampler.sample_img2img(self, samples, noise, conditioning, unconditional_conditioning, steps=self.hr_second_pass_steps or self.steps, image_conditioning=image_conditioning)

			self.is_hr_pass = False

	return samples		return samples