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

---
 modules_forge/main_entry.py | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py
index 36d0a117..e98449b6 100644
--- a/modules_forge/main_entry.py
+++ b/modules_forge/main_entry.py
@@ -5,13 +5,13 @@ from modules import sd_vae as sd_vae_module
 from modules_forge import main_thread
 
 
-sd_model_checkpoint: gr.Dropdown = None
-sd_vae: gr.Dropdown = None
-CLIP_stop_at_last_layers: gr.Slider = None
+ui_checkpoint: gr.Dropdown = None
+ui_vae: gr.Dropdown = None
+ui_clip_skip: gr.Slider = None
 
 
 def make_checkpoint_manager_ui():
-    global sd_model_checkpoint, sd_vae, CLIP_stop_at_last_layers
+    global ui_checkpoint, ui_vae, ui_clip_skip
 
     if shared.opts.sd_model_checkpoint in [None, 'None', 'none', '']:
         if len(sd_models.checkpoints_list) == 0:
@@ -20,22 +20,22 @@ def make_checkpoint_manager_ui():
             shared.opts.set('sd_model_checkpoint', next(iter(sd_models.checkpoints_list.keys())))
 
     sd_model_checkpoint_args = lambda: {"choices": shared_items.list_checkpoint_tiles(shared.opts.sd_checkpoint_dropdown_use_short)}
-    sd_model_checkpoint = gr.Dropdown(
+    ui_checkpoint = gr.Dropdown(
         value=shared.opts.sd_model_checkpoint,
         label="Checkpoint",
         **sd_model_checkpoint_args()
     )
-    ui_common.create_refresh_button(sd_model_checkpoint, shared_items.refresh_checkpoints, sd_model_checkpoint_args, f"forge_refresh_checkpoint")
+    ui_common.create_refresh_button(ui_checkpoint, shared_items.refresh_checkpoints, sd_model_checkpoint_args, f"forge_refresh_checkpoint")
 
     sd_vae_args = lambda: {"choices": shared_items.sd_vae_items()}
-    sd_vae = gr.Dropdown(
+    ui_vae = gr.Dropdown(
         value="Automatic",
         label="VAE",
         **sd_vae_args()
     )
-    ui_common.create_refresh_button(sd_vae, shared_items.refresh_vae_list, sd_vae_args, f"forge_refresh_vae")
+    ui_common.create_refresh_button(ui_vae, shared_items.refresh_vae_list, sd_vae_args, f"forge_refresh_vae")
 
-    CLIP_stop_at_last_layers = gr.Slider(label="Clip skip", value=shared.opts.CLIP_stop_at_last_layers, **{"minimum": 1, "maximum": 12, "step": 1})
+    ui_clip_skip = gr.Slider(label="Clip skip", value=shared.opts.CLIP_stop_at_last_layers, **{"minimum": 1, "maximum": 12, "step": 1})
 
     return
 
@@ -63,7 +63,7 @@ def clip_skip_change(clip_skip):
 
 
 def forge_main_entry():
-    sd_model_checkpoint.change(lambda x: main_thread.async_run(checkpoint_change, x), inputs=[sd_model_checkpoint], show_progress=False)
-    sd_vae.change(lambda x: main_thread.async_run(vae_change, x), inputs=[sd_vae], show_progress=False)
-    CLIP_stop_at_last_layers.change(lambda x: main_thread.async_run(clip_skip_change, x), inputs=[CLIP_stop_at_last_layers], show_progress=False)
+    ui_checkpoint.change(lambda x: main_thread.async_run(checkpoint_change, x), inputs=[ui_checkpoint], show_progress=False)
+    ui_vae.change(lambda x: main_thread.async_run(vae_change, x), inputs=[ui_vae], show_progress=False)
+    ui_clip_skip.change(lambda x: main_thread.async_run(clip_skip_change, x), inputs=[ui_clip_skip], show_progress=False)
     return