Update README.md
Browse filesfixed the model name
README.md
CHANGED
@@ -39,7 +39,7 @@ pipe_prior.to("cuda")
|
|
39 |
prompt = "a hat"
|
40 |
prior_output = pipe_prior(prompt)
|
41 |
|
42 |
-
pipe = KandinskyV22InpaintPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-inpaint", torch_dtype=torch.float16)
|
43 |
pipe.to("cuda")
|
44 |
|
45 |
init_image = load_image(
|
|
|
39 |
prompt = "a hat"
|
40 |
prior_output = pipe_prior(prompt)
|
41 |
|
42 |
+
pipe = KandinskyV22InpaintPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder-inpaint", torch_dtype=torch.float16)
|
43 |
pipe.to("cuda")
|
44 |
|
45 |
init_image = load_image(
|