sdafd commited on
Commit
0293a9d
·
verified ·
1 Parent(s): d424f9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
15
  revision="fp16",
16
  torch_dtype=torch.float16,
17
  )
 
18
 
19
  def inpaint_image(image, mask, prompt, negative_prompt):
20
  # Convert PIL images to numpy arrays if needed
 
15
  revision="fp16",
16
  torch_dtype=torch.float16,
17
  )
18
+ pipe.to('cpu')
19
 
20
  def inpaint_image(image, mask, prompt, negative_prompt):
21
  # Convert PIL images to numpy arrays if needed