Update app.py

app.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 """
 Gradio demo for Wan2.1 First-Last-Frame-to-Video (FLF2V)
-Uses Accelerate’s automatic device mapping for optimal CPU/GPU placement.
+Uses Accelerate’s balanced device mapping for optimal CPU/GPU placement.
 Author: <your-handle>
 """
 
@@ -39,7 +39,7 @@ def load_pipeline():
         image_encoder=image_encoder,
         torch_dtype=DTYPE,
         low_cpu_mem_usage=True,   # lazy-load weights into CPU RAM
-        device_map="auto",        # auto-split across CPU/GPU
+        device_map="balanced",    # balanced CPU/GPU sharding
     )
 
     # 4) use the fast Rust-backed processor