-
Notifications
You must be signed in to change notification settings - Fork 178
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Callback for Text2Image Pipeline, Flux pipeline fix #1223
Callback for Text2Image Pipeline, Flux pipeline fix #1223
Conversation
8ab70eb
to
4b3e0b6
Compare
e866b0e
to
8aa3350
Compare
4bfd3fc
to
0bf9a85
Compare
0bf9a85
to
ac79ae8
Compare
(text2image_generate_docstring + std::string(" \n ")).c_str() | ||
); | ||
(text2image_generate_docstring + std::string(" \n ")).c_str()) | ||
.def("decode", &ov::genai::Text2ImagePipeline::decode, py::arg("latent"));; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
is there any plans also demonstrate decode method usage somewhere in readme?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
} | ||
|
||
latents = unpack_latents(latents, generation_config.height, generation_config.width, vae_scale_factor); | ||
return m_vae->decode(latents); | ||
} | ||
|
||
ov::Tensor decode(const ov::Tensor latent) override { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
looks like it should also perform unpack_latents
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Callback sample with decoding: