feylur commited on
Commit
853b9df
·
verified ·
1 Parent(s): ecba289

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def load_models():
40
  base_ckpt="booksforcharlie/stable-diffusion-inpainting",
41
  attn_ckpt=repo_path,
42
  attn_ckpt_version="mix",
43
- weight_dtype=init_weight_dtype("fp32"), # CPU uses fp32
44
  use_tf32=False, # CPU doesn't support TF32
45
  device='cpu'
46
  )
 
40
  base_ckpt="booksforcharlie/stable-diffusion-inpainting",
41
  attn_ckpt=repo_path,
42
  attn_ckpt_version="mix",
43
+ weight_dtype=torch.float32, # CPU uses fp32
44
  use_tf32=False, # CPU doesn't support TF32
45
  device='cpu'
46
  )