Update app.py

app.py

@@ -314,6 +314,11 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
 
     selected_loras = [loras_state[idx] for idx in selected_indices]
 
+    base_weight_param = transformer_state_dict.get(base_param_name, None)
+    if base_weight_param is None:
+        print(f"Warning: {base_param_name} not found in transformer_state_dict.")
+
+
     # Build the prompt with trigger words
     prepends = []
     appends = []