FLUXPro

Runtime error

KingNish commited on Oct 4, 2024

Commit

0fe7f63

verified ·

1 Parent(s): e386106

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ DEFAULT_INFERENCE_STEPS = 1
 # Device and model setup
 dtype = torch.float16
 pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
-    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
 ).to("cuda")
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 # pipe.transformer.to(memory_format=torch.channels_last)

 # Device and model setup
 dtype = torch.float16
 pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
+    "ostris/OpenFLUX.1", torch_dtype=dtype
 ).to("cuda")
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 # pipe.transformer.to(memory_format=torch.channels_last)