From 7428fb5176ccfd203bddcfa30d75c8df5a772cb4 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 29 Apr 2023 16:28:51 +0300
Subject: add is_hr_pass field for processing

---
 modules/processing.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index c50784f4..8d7b2462 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -164,6 +164,7 @@ class StableDiffusionProcessing:
         self.all_seeds = None
         self.all_subseeds = None
         self.iteration = 0
+        self.is_hr_pass = False
         
 
     @property
@@ -883,6 +884,8 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
         if not self.enable_hr:
             return samples
 
+        self.is_hr_pass = True
+
         target_width = self.hr_upscale_to_x
         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)
 
+        self.is_hr_pass = False
+
         return samples
 
 
-- 
cgit v1.2.1