yizhangliu commited on
Commit
4a2684c
1 Parent(s): b4fd607

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -284,9 +284,9 @@ def load_sd_model():
284
  if os.environ.get('IS_MY_DEBUG') is None:
285
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
286
  "runwayml/stable-diffusion-inpainting",
287
- # revision="fp16",
288
  # "stabilityai/stable-diffusion-2-inpainting",
289
- # torch_dtype=torch.float16,
290
  )
291
  sd_pipe = sd_pipe.to(device)
292
 
 
284
  if os.environ.get('IS_MY_DEBUG') is None:
285
  sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
286
  "runwayml/stable-diffusion-inpainting",
287
+ revision="fp16",
288
  # "stabilityai/stable-diffusion-2-inpainting",
289
+ torch_dtype=torch.float16,
290
  )
291
  sd_pipe = sd_pipe.to(device)
292