From fa9370b7411166c19e8e386400dc4e6082f47b2d Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Sun, 13 Aug 2023 06:07:30 +0300
Subject: add refiner to StableDiffusionProcessing class write out correct
 model name in infotext, rather than the refiner model

---
 modules/processing_scripts/refiner.py | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

(limited to 'modules/processing_scripts')

diff --git a/modules/processing_scripts/refiner.py b/modules/processing_scripts/refiner.py
index 773ec5d0..7b946d05 100644
--- a/modules/processing_scripts/refiner.py
+++ b/modules/processing_scripts/refiner.py
@@ -41,15 +41,9 @@ class ScriptRefiner(scripts.Script):
     def before_process(self, p, enable_refiner, refiner_checkpoint, refiner_switch_at):
         # the actual implementation is in sd_samplers_common.py, apply_refiner
 
-        p.refiner_checkpoint_info = None
-        p.refiner_switch_at = None
-
         if not enable_refiner or refiner_checkpoint in (None, "", "None"):
-            return
-
-        refiner_checkpoint_info = sd_models.get_closet_checkpoint_match(refiner_checkpoint)
-        if refiner_checkpoint_info is None:
-            raise Exception(f'Could not find checkpoint with name {refiner_checkpoint}')
-
-        p.refiner_checkpoint_info = refiner_checkpoint_info
-        p.refiner_switch_at = refiner_switch_at
+            p.refiner_checkpoint_info = None
+            p.refiner_switch_at = None
+        else:
+            p.refiner_checkpoint = refiner_checkpoint
+            p.refiner_switch_at = refiner_switch_at
-- 
cgit v1.2.3