Spaces:

Ryukijano
/

Fastest-image-generation

Sleeping

Ryukijano commited on Dec 9, 2024

Commit

3567671

verified ·

1 Parent(s): cd3c865

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ DEFAULT_INFERENCE_STEPS = 1
 # Device and model setup
 dtype = torch.bfloat16
 pipe = FluxWithCFGPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, use_safetensors=True, variant="fp16"
 )
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype, use_safetensors=True, variant="fp16")
 pipe.to("cuda")

 # Device and model setup
 dtype = torch.bfloat16
 pipe = FluxWithCFGPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, use_safetensors=True
 )
 pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype, use_safetensors=True, variant="fp16")
 pipe.to("cuda")