YiYiXu sayakpaul HF staff commited on
Commit
6745411
1 Parent(s): 41cf98f

Update README.md (#1)

Browse files

- Update README.md (d22c4c20bd5bd233e28e2d233d3245cc16ab6052)


Co-authored-by: Sayak Paul <sayakpaul@users.noreply.huggingface.co>

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(