Update README.md (#2)
Browse files- Update README.md (6fb86d35212ce498c7125d697fcd3541cdec5a62)
Co-authored-by: Shayani <[email protected]>
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(
|