jiuface commited on
Commit
df50fe7
·
verified ·
1 Parent(s): 0bf16dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ login(token=HF_TOKEN)
27
  import diffusers
28
 
29
  # init
30
- dtype = torch.float16 # use float16 for fast generate
31
  device = "cuda" if torch.cuda.is_available() else "cpu"
32
  base_model = "black-forest-labs/FLUX.1-dev"
33
 
 
27
  import diffusers
28
 
29
  # init
30
+ dtype = torch.bfloat16
31
  device = "cuda" if torch.cuda.is_available() else "cpu"
32
  base_model = "black-forest-labs/FLUX.1-dev"
33