liuyizhang commited on
Commit
e4af7ee
1 Parent(s): 7273b98

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -250,9 +250,9 @@ logger.info(f"initialize stable-diffusion-inpainting...")
250
  sd_pipe = None
251
  if os.environ.get('IS_MY_DEBUG') is None:
252
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
253
- # "runwayml/stable-diffusion-inpainting",
254
  # revision="fp16"
255
- "stabilityai/stable-diffusion-2-inpainting",
256
  torch_dtype=torch.float16
257
  )
258
  sd_pipe = sd_pipe.to(device)
 
250
  sd_pipe = None
251
  if os.environ.get('IS_MY_DEBUG') is None:
252
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
253
+ "runwayml/stable-diffusion-inpainting",
254
  # revision="fp16"
255
+ # "stabilityai/stable-diffusion-2-inpainting",
256
  torch_dtype=torch.float16
257
  )
258
  sd_pipe = sd_pipe.to(device)