From b4ca5d742047db16822a976b683600222661f9ea Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Tue, 6 Aug 2024 14:38:40 -0700
Subject: [PATCH] model load entry for all model loads from webui

including startup, checkpoint selection, and that HTML tab of some preview images
---
 modules_forge/main_entry.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py
index 689eb1b9..9735883a 100644
--- a/modules_forge/main_entry.py
+++ b/modules_forge/main_entry.py
@@ -40,12 +40,17 @@ def make_checkpoint_manager_ui():
     return
 
 
+def model_load_entry():
+    sd_models.load_model()
+    return
+
+
 def checkpoint_change(ckpt_name):
     print(f'Checkpoint Selected: {ckpt_name}')
     shared.opts.set('sd_model_checkpoint', ckpt_name)
     shared.opts.save(shared.config_filename)
 
-    sd_models.load_model()
+    model_load_entry()
     return
 
 
@@ -68,5 +73,5 @@ def forge_main_entry():
     ui_clip_skip.change(lambda x: main_thread.async_run(clip_skip_change, x), inputs=[ui_clip_skip], show_progress=False)
 
     # Load Model
-    main_thread.async_run(sd_models.load_model)
+    main_thread.async_run(model_load_entry)
     return