sdafd commited on
Commit
e4b23b2
·
verified ·
1 Parent(s): 4d956f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ safety_checker.StableDiffusionSafetyChecker.forward = sc
12
 
13
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
14
  "runwayml/stable-diffusion-inpainting",
15
- revision="fp32",
16
- torch_dtype=torch.float32,
17
  )
18
  pipe.to('cpu')
19
 
 
12
 
13
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
14
  "runwayml/stable-diffusion-inpainting",
15
+ revision="fp16",
16
+ torch_dtype=torch.float16,
17
  )
18
  pipe.to('cpu')
19