sayakpaul HF staff commited on
Commit
d22c4c2
1 Parent(s): 3431784

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -31,13 +31,13 @@ from diffusers.utils import load_image
31
  import torch
32
  import numpy as np
33
 
34
- pipe_prior = KandinskyPriorPipeline.from_pretrained("YiYiXu/Kandinsky-prior", torch_dtype=torch.float16)
35
  pipe_prior.to("cuda")
36
 
37
  prompt = "a hat"
38
  image_emb, zero_image_emb = pipe_prior(prompt, return_dict=False)
39
 
40
- pipe = KandinskyInpaintPipeline.from_pretrained("YiYiXu/Kandinsky-inpaint", torch_dtype=torch.float16)
41
  pipe.to("cuda")
42
 
43
  init_image = load_image(
 
31
  import torch
32
  import numpy as np
33
 
34
+ pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16)
35
  pipe_prior.to("cuda")
36
 
37
  prompt = "a hat"
38
  image_emb, zero_image_emb = pipe_prior(prompt, return_dict=False)
39
 
40
+ pipe = KandinskyInpaintPipeline.from_pretrained("kandinsky-community/kandinsky-2-1-inpaint", torch_dtype=torch.float16)
41
  pipe.to("cuda")
42
 
43
  init_image = load_image(