Update README.md
Browse files
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("
|
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("
|
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(
|