sayakpaul HF staff commited on
Commit
f3d1fe8
1 Parent(s): 8d6a880

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -29,7 +29,7 @@ from diffusers import KandinskyPipeline, KandinskyPriorPipeline
29
  import torch
30
 
31
 
32
- pipe_prior = KandinskyPriorPipeline.from_pretrained("YiYiXu/Kandinsky-prior", torch_dtype=torch.float16)
33
  pipe_prior.to("cuda")
34
 
35
  prompt = "A alien cheeseburger creature eating itself, claymation, cinematic, moody lighting"
@@ -43,7 +43,7 @@ zero_image_emb = pipe_prior(
43
  negative_prompt, guidance_scale=1.0, num_inference_steps=25, generator=generator, negative_prompt=negative_prompt
44
  ).images
45
 
46
- pipe = KandinskyPipeline.from_pretrained("YiYiXu/Kandinsky", torch_dtype=torch.float16)
47
  pipe.to("cuda")
48
 
49
 
@@ -81,11 +81,11 @@ original_image = Image.open(BytesIO(response.content)).convert("RGB")
81
  original_image = original_image.resize((768, 512))
82
 
83
  # create prior
84
- pipe_prior = KandinskyPriorPipeline.from_pretrained("YiYiXu/Kandinsky-prior", torch_dtype=torch.float16)
85
  pipe_prior.to("cuda")
86
 
87
  # create img2img pipeline
88
- pipe = KandinskyImg2ImgPipeline.from_pretrained("YiYiXu/Kandinsky", torch_dtype=torch.float16)
89
  pipe.to("cuda")
90
 
91
  prompt = "A fantasy landscape, Cinematic lighting"
@@ -126,7 +126,7 @@ import PIL
126
  import torch
127
  from torchvision import transforms
128
 
129
- pipe_prior = KandinskyPriorPipeline.from_pretrained("YiYiXu/Kandinsky-prior", torch_dtype=torch.float16)
130
  pipe_prior.to("cuda")
131
 
132
  img1 = load_image(
@@ -141,7 +141,7 @@ images_texts = ["a cat", img1, img2]
141
  weights = [0.3, 0.3, 0.4]
142
  image_emb, zero_image_emb = pipe_prior.interpolate(images_texts, weights)
143
 
144
- pipe = KandinskyPipeline.from_pretrained("YiYiXu/Kandinsky", torch_dtype=torch.float16)
145
  pipe.to("cuda")
146
 
147
  image = pipe(
 
29
  import torch
30
 
31
 
32
+ pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16)
33
  pipe_prior.to("cuda")
34
 
35
  prompt = "A alien cheeseburger creature eating itself, claymation, cinematic, moody lighting"
 
43
  negative_prompt, guidance_scale=1.0, num_inference_steps=25, generator=generator, negative_prompt=negative_prompt
44
  ).images
45
 
46
+ pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1", torch_dtype=torch.float16)
47
  pipe.to("cuda")
48
 
49
 
 
81
  original_image = original_image.resize((768, 512))
82
 
83
  # create prior
84
+ pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16)
85
  pipe_prior.to("cuda")
86
 
87
  # create img2img pipeline
88
+ pipe = KandinskyImg2ImgPipeline.from_pretrained("kandinsky-community/kandinsky-2-1", torch_dtype=torch.float16)
89
  pipe.to("cuda")
90
 
91
  prompt = "A fantasy landscape, Cinematic lighting"
 
126
  import torch
127
  from torchvision import transforms
128
 
129
+ pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16)
130
  pipe_prior.to("cuda")
131
 
132
  img1 = load_image(
 
141
  weights = [0.3, 0.3, 0.4]
142
  image_emb, zero_image_emb = pipe_prior.interpolate(images_texts, weights)
143
 
144
+ pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1", torch_dtype=torch.float16)
145
  pipe.to("cuda")
146
 
147
  image = pipe(