Upload 2 files

app.py

@@ -85,6 +85,19 @@ def change_base_model(repo_id: str, cn_on: bool):
 
 change_base_model.zerogpu = True
 
+def preload_base_model(repo_id: str, cn_on: bool, progress=gr.Progress(track_tqdm=True)):
+    global last_model
+    global last_cn_on
+    try:
+        if (repo_id == last_model and cn_on is last_cn_on) or not is_repo_name(repo_id) or not is_repo_exists(repo_id): yield gr.update(visible=True)
+        DiffusionPipeline.from_pretrained(repo_id, torch_dtype=dtype)
+        print(f"Caching Model")
+    except Exception as e:
+        print(f"Model cache Error: {e}")
+    yield gr.update(visible=True)
+
+preload_base_model.zerogpu = True
+
 class calculateDuration:
     def __init__(self, activity_name=""):
         self.activity_name = activity_name
@@ -502,7 +515,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
     deselect_lora_button.click(deselect_lora, None, [prompt, selected_info, selected_index, width, height], queue=False, show_api=False)
     gr.on(
         triggers=[model_name.change, cn_on.change],
-        fn=change_base_model,
+        fn=preload_base_model,
         inputs=[model_name, cn_on],
         outputs=[result],
         queue=True,