Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
| 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 |
)
|