Change of model path (#5)
Browse files- Change of model path (7e2501ebff8a9573f0f8bf854984da8a7f8c0666)
Co-authored-by: Markus <[email protected]>
README.md
CHANGED
@@ -27,7 +27,7 @@ from diffusers import AutoPipelineForInpainting
|
|
27 |
from diffusers.utils import load_image
|
28 |
import torch
|
29 |
|
30 |
-
pipe = AutoPipelineForInpainting.from_pretrained("
|
31 |
|
32 |
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
|
33 |
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
|
|
|
27 |
from diffusers.utils import load_image
|
28 |
import torch
|
29 |
|
30 |
+
pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
|
31 |
|
32 |
img_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
|
33 |
mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
|