From 4122e7f239e48b9affc8222c2ab9b037a5db30cb Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Tue, 6 Aug 2024 14:07:04 -0700
Subject: [PATCH] change name

---
 modules/ui.py          | 2 +-
 modules/ui_settings.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/modules/ui.py b/modules/ui.py
index 3f2455c3..aa647b96 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -942,7 +942,7 @@ def create_ui():
         settings.text_settings.change(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])
         demo.load(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])
 
-        modelmerger_ui.setup_ui(dummy_component=dummy_component, sd_model_checkpoint_component=main_entry.sd_model_checkpoint)
+        modelmerger_ui.setup_ui(dummy_component=dummy_component, sd_model_checkpoint_component=main_entry.ui_checkpoint)
 
         main_entry.forge_main_entry()
 
diff --git a/modules/ui_settings.py b/modules/ui_settings.py
index 1d5c504c..bc9c8296 100644
--- a/modules/ui_settings.py
+++ b/modules/ui_settings.py
@@ -330,8 +330,8 @@ class UiSettings:
         button_set_checkpoint.click(
             fn=button_set_checkpoint_change,
             _js="function(v){ var res = desiredCheckpointName; desiredCheckpointName = ''; return [res || v, null]; }",
-            inputs=[main_entry.sd_model_checkpoint, self.dummy_component],
-            outputs=[main_entry.sd_model_checkpoint, self.text_settings],
+            inputs=[main_entry.ui_checkpoint, self.dummy_component],
+            outputs=[main_entry.ui_checkpoint, self.text_settings],
         )
 
         component_keys = [k for k in opts.data_labels.keys() if k in self.component_dict]