Astridkraft commited on
Commit
48441b2
·
verified ·
1 Parent(s): 248bdf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ def load_img2img():
72
  if pipe_img2img is None:
73
  print("Loading Image-to-Image model...")
74
  pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
75
- "runwayml/stable-diffusion-v1-5",
76
  torch_dtype=torch_dtype,
77
  use_safetensors=True,
78
  safety_checker=None,
 
72
  if pipe_img2img is None:
73
  print("Loading Image-to-Image model...")
74
  pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
75
+ "runwayml/stable-diffusion-inpainting",
76
  torch_dtype=torch_dtype,
77
  use_safetensors=True,
78
  safety_checker=None,