jadechoghari commited on
Commit
33743fb
Β·
verified Β·
1 Parent(s): f57b79f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ from torchao.quantization import autoquant
15
  # pipeline_normal.transformer = torch.compile(pipeline_normal.transformer, mode="max-autotune", fullgraph=True)
16
 
17
  pipeline_normal = DiffusionPipeline.from_pretrained("sayakpaul/FLUX.1-merged")
 
18
  pipeline_normal.load_lora_weights("DarkMoonDragon/TurboRender-flux-dev")
19
  # # optimized FluxPipeline
20
  # pipeline_optimized = FluxPipeline.from_pretrained(
 
15
  # pipeline_normal.transformer = torch.compile(pipeline_normal.transformer, mode="max-autotune", fullgraph=True)
16
 
17
  pipeline_normal = DiffusionPipeline.from_pretrained("sayakpaul/FLUX.1-merged")
18
+ pipeline_normal.enable_model_cpu_offload()
19
  pipeline_normal.load_lora_weights("DarkMoonDragon/TurboRender-flux-dev")
20
  # # optimized FluxPipeline
21
  # pipeline_optimized = FluxPipeline.from_pretrained(