File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed
src/diffusers/pipelines/kandinsky Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change 41
41
>>> from diffusers import KandinskyPipeline, KandinskyPriorPipeline
42
42
>>> import torch
43
43
44
- >>> pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/Kandinsky-prior")
44
+ >>> pipe_prior = KandinskyPriorPipeline.from_pretrained("kandinsky-community/Kandinsky-2-1- prior")
45
45
>>> pipe_prior.to("cuda")
46
46
47
47
>>> prompt = "red cat, 4k photo"
48
48
>>> out = pipe_prior(prompt)
49
- >>> image_emb = out.images
50
- >>> zero_image_emb = out.zero_embeds
49
+ >>> image_emb = out.image_embeds
50
+ >>> negative_image_emb = out.negative_image_embeds
51
51
52
52
>>> pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1")
53
53
>>> pipe.to("cuda")
54
54
55
55
>>> image = pipe(
56
56
... prompt,
57
57
... image_embeds=image_emb,
58
- ... negative_image_embeds=zero_image_emb ,
58
+ ... negative_image_embeds=negative_image_emb ,
59
59
... height=768,
60
60
... width=768,
61
61
... num_inference_steps=100,
Original file line number Diff line number Diff line change 45
45
46
46
>>> prompt = "red cat, 4k photo"
47
47
>>> out = pipe_prior(prompt)
48
- >>> image_emb = out.images
49
- >>> zero_image_emb = out.zero_embeds
48
+ >>> image_emb = out.image_embeds
49
+ >>> negative_image_emb = out.negative_image_embeds
50
50
51
51
>>> pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1")
52
52
>>> pipe.to("cuda")
53
53
54
54
>>> image = pipe(
55
55
... prompt,
56
56
... image_embeds=image_emb,
57
- ... negative_image_embeds=zero_image_emb ,
57
+ ... negative_image_embeds=negative_image_emb ,
58
58
... height=768,
59
59
... width=768,
60
60
... num_inference_steps=100,
You can’t perform that action at this time.
0 commit comments