Spaces:

jadechoghari
/

flux-kiwi

Runtime error

jadechoghari commited on Sep 14, 2024

Commit

bceeac1

verified ·

1 Parent(s): 435d513

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from torchao.quantization import autoquant
 # optimized FluxPipeline
 pipeline_optimized = FluxPipeline.from_pretrained(
-    "sayakpaul/FLUX.1-merged",
     torch_dtype=torch.bfloat16
 ).to("cuda")

 # optimized FluxPipeline
 pipeline_optimized = FluxPipeline.from_pretrained(
+    "camenduru/FLUX.1-dev-diffusers",
     torch_dtype=torch.bfloat16
 ).to("cuda")