Update app.py

app.py

@@ -315,8 +315,9 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
     selected_loras = [loras_state[idx] for idx in selected_indices]
 
     # Initialize transformer_state_dict
-    transformer_state_dict = pipe.transformer.state_dict()  # Ensure this is correctly defined
-
+        transformer_state_dict = pipe.transformer.state_dict()  # Ensure this is correctly defined
+    
+    base_param_name = "weight"
     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.")