sdafd commited on
Commit
c865610
·
verified ·
1 Parent(s): 4913982

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ safety_checker.StableDiffusionSafetyChecker.forward = sc
12
 
13
  # Initialize the inpainting pipeline
14
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
15
- "ckpt/dreamlike-diffusion-1.0-inpainting"
 
16
  )
17
  pipe.to('cpu')
18
 
 
12
 
13
  # Initialize the inpainting pipeline
14
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
15
+ "stabilityai/stable-diffusion-2-inpainting",
16
+ torch_dtype=torch.float32,
17
  )
18
  pipe.to('cpu')
19