Spaces:

multimodalart
/

stable-diffusion-inpainting

Sleeping

multimodalart HF Staff commited on 17 days ago

Commit

dc429f8

verified ·

1 Parent(s): 9fdb553

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,8 +23,7 @@ def download_image(url):
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting",
-    revision="fp16",
     torch_dtype=torch.float16,
     use_auth_token=auth_token,
 ).to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained(
+    "stable-diffusion-v1-5/stable-diffusion-v1-5",
     torch_dtype=torch.float16,
     use_auth_token=auth_token,
 ).to(device)