Hugging Face's logo

Spaces:
Ryukijano
/
Fastest-image-generation
Running on Zero

App Files Community
Ryukijano commited on
Commit
cd3c865
·
verified ·
1 Parent(s): b5a0af4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ DEFAULT_HEIGHT = 1024
19
  DEFAULT_INFERENCE_STEPS = 1
20
 
21
  # Device and model setup
22
- dtype = torch.float16
23
  pipe = FluxWithCFGPipeline.from_pretrained(
24
  "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, use_safetensors=True, variant="fp16"
25
  )
 
19
  DEFAULT_INFERENCE_STEPS = 1
20
 
21
  # Device and model setup
22
+ dtype = torch.bfloat16
23
  pipe = FluxWithCFGPipeline.from_pretrained(
24
  "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, use_safetensors=True, variant="fp16"
25
  )