banyapon commited on
Commit
c84e3c5
·
1 Parent(s): 3b26803

checkpoint 360 app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ except Exception as e:
20
 
21
  # Load Inpainting Model (Without CUDA)
22
  try:
23
- #pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting")
24
- pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-inpainting-cpu", torch_dtype=torch.float32)
25
  except Exception as e:
26
  raise RuntimeError(f"Error loading inpainting model: {e}")
27
 
 
20
 
21
  # Load Inpainting Model (Without CUDA)
22
  try:
23
+ pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting")
24
+
25
  except Exception as e:
26
  raise RuntimeError(f"Error loading inpainting model: {e}")
27