LPX55 commited on
Commit
453b4ed
·
verified ·
1 Parent(s): 0c8e08c

Update optimized.py

Browse files
Files changed (1) hide show
  1. 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