liuyizhang commited on
Commit
70c927a
1 Parent(s): c7d2050

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -246,7 +246,8 @@ logger.info(f"initialize stable-diffusion-inpainting...")
246
  sd_pipe = None
247
  if os.environ.get('IS_MY_DEBUG') is None:
248
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
249
- "runwayml/stable-diffusion-inpainting",
 
250
  torch_dtype=torch.float16
251
  )
252
  sd_pipe = sd_pipe.to(device)
 
246
  sd_pipe = None
247
  if os.environ.get('IS_MY_DEBUG') is None:
248
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
249
+ # "runwayml/stable-diffusion-inpainting",
250
+ "stabilityai/stable-diffusion-2-inpainting",
251
  torch_dtype=torch.float16
252
  )
253
  sd_pipe = sd_pipe.to(device)