Spaces:
Yuanshi
/
Running on Zero

Yuanshi commited on
Commit
561cdae
·
verified ·
1 Parent(s): cdbe05f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ pipe = FluxPipeline.from_pretrained(bfl_repo, transformer=None, torch_dtype=dtyp
20
  pipe.transformer = transformer
21
  pipe.scheduler.config.use_dynamic_shifting = False
22
  pipe.scheduler.config.time_shift = 10
23
- # pipe.enable_model_cpu_offload()
24
  pipe = pipe.to(device)
25
 
26
  pipe.load_lora_weights(
 
20
  pipe.transformer = transformer
21
  pipe.scheduler.config.use_dynamic_shifting = False
22
  pipe.scheduler.config.time_shift = 10
23
+ pipe.enable_model_cpu_offload()
24
  pipe = pipe.to(device)
25
 
26
  pipe.load_lora_weights(