Update README.md

#2
by Hooman - opened
Files changed (1) hide show
  1. README.md +1 -1
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(