alfredplpl commited on
Commit
ebceffd
·
verified ·
1 Parent(s): d5f8284

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ import os
19
  device="cuda"
20
  shape=(1,48//4,16,256//8,256//8)
21
  sample_N=25
22
- torch_dtype=torch.bfloat16
23
  eps=1
24
  cfg=2.5
25
 
 
19
  device="cuda"
20
  shape=(1,48//4,16,256//8,256//8)
21
  sample_N=25
22
+ torch_dtype=torch.float32
23
  eps=1
24
  cfg=2.5
25