Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
| 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,
|