Update app.py

app.py

@@ -314,11 +314,13 @@ 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
+
     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 = []