Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- optimized.py +1 -1
optimized.py
CHANGED
|
@@ -61,7 +61,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
|
|
| 61 |
pipe.to(device)
|
| 62 |
# 3. Strict Order for Optimization Steps
|
| 63 |
# A. Apply CPU Offloading FIRST
|
| 64 |
-
pipe.enable_sequential_cpu_offload() # No arguments for new API
|
| 65 |
# 2. Then apply custom VAE slicing
|
| 66 |
if getattr(pipe, "vae", None) is not None:
|
| 67 |
# Method 1: Use official implementation if available
|
|
|
|
| 61 |
pipe.to(device)
|
| 62 |
# 3. Strict Order for Optimization Steps
|
| 63 |
# A. Apply CPU Offloading FIRST
|
| 64 |
+
#### pipe.enable_sequential_cpu_offload() # No arguments for new API
|
| 65 |
# 2. Then apply custom VAE slicing
|
| 66 |
if getattr(pipe, "vae", None) is not None:
|
| 67 |
# Method 1: Use official implementation if available
|