Hugging Face's logo

Spaces:
gokaygokay
/
FLUX.1-dev-with-Captioner
Running on Zero

App Files Community
3
gokaygokay commited on
Commit
2e5c176
·
verified ·
1 Parent(s): b4b6aab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ from pipelines import FluxPipeline
31
 
32
  pipe = FluxPipeline.from_pretrained(
33
  "black-forest-labs/FLUX.1-dev",
34
- torch_dtype=torch.float32, # Load in full precision initially
35
  vae=taef1,
36
  token=huggingface_token
37
  ).to(device)
 
31
 
32
  pipe = FluxPipeline.from_pretrained(
33
  "black-forest-labs/FLUX.1-dev",
34
+ torch_dtype=dtype,
35
  vae=taef1,
36
  token=huggingface_token
37
  ).to(device)