kandinsky-community
/

kandinsky-2-2-decoder-inpaint

KandinskyV22InpaintPipeline

Model card Files Files and versions

Update README.md

#2

by Hooman - opened Jul 14, 2023

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ pipe_prior.to("cuda")
 prompt = "a hat"
 prior_output = pipe_prior(prompt)
-pipe = KandinskyV22InpaintPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-inpaint", torch_dtype=torch.float16)
 pipe.to("cuda")
 init_image = load_image(

 prompt = "a hat"
 prior_output = pipe_prior(prompt)
+pipe = KandinskyV22InpaintPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder-inpaint", torch_dtype=torch.float16)
 pipe.to("cuda")
 init_image = load_image(