Update app.py
Browse files
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 |
-
|
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(
|