KheemDH commited on
Commit
f881321
Β·
1 Parent(s): 6d80f3d

Change model precision to float32 for CPU compatibility

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="fp32", # Use fp32 for CPU
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