Merge pull request #169 from ototadana/fix/refiner_switch_at-error

fix use_refiner_model_only error
ototadana · Oct 29, 2023 · 7ef2935 · 7ef2935
2 parents 9c90c36 + 95c41ad
commit 7ef2935
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scripts/inferencers/img2img_face_processor.py b/scripts/inferencers/img2img_face_processor.py
@@ -32,13 +32,14 @@ def process(
         if len(np) > 0:
             p.negative_prompt = np
 
-        refiner_switch_at = p.refiner_switch_at
         if use_refiner_model_only:
+            refiner_switch_at = p.refiner_switch_at
             p.refiner_switch_at = 0
 
         print(f"prompt for the {face.face_area.tag}: {p.prompt}")
 
         proc = process_images(p)
-        p.refiner_switch_at = refiner_switch_at
+        if use_refiner_model_only:
+            p.refiner_switch_at = refiner_switch_at
 
         return proc.images[0]