Update app.py

app.py

@@ -423,6 +423,9 @@ def apply_current_loras_to_transformer(
 ):
     global _transformer
 
+    # Get device from the pipeline (fallback: first parameter of transformer)
+    device = getattr(pipeline, 'device', None) or next(_transformer.parameters()).device
+
     # Collect non-zero strength LoRAs
     lora_specs = _collect_lora_specs(
         pose_strength, general_strength, motion_strength, dreamlay_strength,
@@ -465,7 +468,7 @@ def apply_current_loras_to_transformer(
     # Load fused state dict into transformer
     with torch.no_grad():
         fused_state_cuda = {
-            k: (v.to(_transformer.device) if v.device == torch.device("cpu") else v)
+            k: (v.to(device) if v.device == torch.device("cpu") else v)
             for k, v in fused_state.items()
         }
         missing, unexpected = _transformer.load_state_dict(fused_state_cuda, strict=False)