Change model precision to float32 for CPU compatibility
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ else:
|
|
| 26 |
"diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
|
| 27 |
vae=vae,
|
| 28 |
torch_dtype=torch.float32, # Use fp32 for CPU
|
| 29 |
-
variant="
|
| 30 |
use_safetensors=True
|
| 31 |
).to(device) # Ensure it uses the CPU if no GPU
|
| 32 |
|
|
|
|
| 26 |
"diffusers/stable-diffusion-xl-1.0-inpainting-0.1",
|
| 27 |
vae=vae,
|
| 28 |
torch_dtype=torch.float32, # Use fp32 for CPU
|
| 29 |
+
variant="fp16", # Use fp32 for CPU
|
| 30 |
use_safetensors=True
|
| 31 |
).to(device) # Ensure it uses the CPU if no GPU
|
| 32 |
|