From 51a3e4a7025bfafdb778986040d5e1864eed9c7a Mon Sep 17 00:00:00 2001
From: Chris Malone <chrismrulz@hotmail.com>
Date: Fri, 21 Mar 2025 11:16:55 +1100
Subject: [PATCH] fix lora loading, finds the indices from filenames

---
 gradio_server.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/gradio_server.py b/gradio_server.py
index a4a3e4b..6647188 100644
--- a/gradio_server.py
+++ b/gradio_server.py
@@ -603,7 +603,16 @@ def load_models(i2v,  lora_dir,  lora_preselected_preset ):
 
     offloadobj = offload.profile(pipe, profile_no= profile, compile = compile, quantizeTransformer = quantizeTransformer, loras = "transformer", **kwargs)  
     loras, loras_names, default_loras_choices, default_loras_multis_str, default_prompt, default_lora_preset, loras_presets = setup_loras(pipe["transformer"],  lora_dir, lora_preselected_preset, None)
-
+    if "activated_loras" in ui_defaults:
+        lora_filenames = [os.path.basename(lora_path) for lora_path in loras]
+        activated_indices = []
+        for lora_file in ui_defaults["activated_loras"]:
+            try:
+                idx = lora_filenames.index(lora_file)
+                activated_indices.append(str(idx))
+            except ValueError:
+                print(f"Warning: Lora file {lora_file} from config not found in loras directory")
+        ui_defaults["activated_loras"] = activated_indices
 
     return wan_model, offloadobj, loras, loras_names, default_loras_choices, default_loras_multis_str, default_prompt, default_lora_preset, loras_presets