Update app.py

app.py

@@ -37,7 +37,7 @@ pipe = DiffusionPipeline.from_pretrained(
     feature_extractor=feature_extractor,
     torch_dtype=torch.bfloat16,
     use_safetensors=True,
-    device_map="auto",          # automatically spreads submodules CPU/GPU
+    device_map="balanced",          # automatically spreads submodules CPU/GPU
     offload_folder="offload"    # where to spill CPU-offloaded weights
 )