Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|