Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
| 22 |
low_cpu_mem_usage=True
|
| 23 |
)
|
| 24 |
# Make sure your HF token has gated access to Flux
|
| 25 |
-
|
| 26 |
pipe.to(device)
|
| 27 |
pipe.enable_model_cpu_offload()
|
| 28 |
|
|
|
|
| 22 |
low_cpu_mem_usage=True
|
| 23 |
)
|
| 24 |
# Make sure your HF token has gated access to Flux
|
| 25 |
+
|
| 26 |
pipe.to(device)
|
| 27 |
pipe.enable_model_cpu_offload()
|
| 28 |
|