Upload 2 files

mod.py

@@ -71,15 +71,17 @@ def get_repo_safetensors(repo_id: str):
 # Initialize the base model
 base_model = models[0]
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-
+last_model = models[0]
 
 def change_base_model(repo_id: str, progress=gr.Progress(track_tqdm=True)):
     global pipe
+    global last_model
     try:
-        if not is_repo_name(repo_id) or not is_repo_exists(repo_id): return
+        if repo_id == last_model or not is_repo_name(repo_id) or not is_repo_exists(repo_id): return
         progress(0, f"Loading model: {repo_id}")
         clear_cache()
         pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
+        last_model = repo_id
         progress(1, f"Model loaded: {repo_id}")
     except Exception as e:
         print(e)
@@ -135,6 +137,7 @@ def fuse_loras(pipe, lorajson: list[dict]):
     #pipe.unload_lora_weights()
 
 
+change_base_model.zerogpu = True
 fuse_loras.zerogpu = True