Unverified Commit 256a33ec authored by Li Wang's avatar Li Wang Committed by GitHub
Browse files

[Model] Fix bagel failed to run (#31132)


Signed-off-by: default avatarwangli <wangli858794774@gmail.com>
parent c02a2705
......@@ -487,7 +487,7 @@ class BagelForConditionalGeneration(
# Split by image
return tuple(vision_embeds)
def get_multimodal_embeddings(self, **kwargs: object) -> MultiModalEmbeddings:
def embed_multimodal(self, **kwargs: object) -> MultiModalEmbeddings:
"""Get multimodal embeddings from input."""
image_input = self._parse_and_validate_image_input(**kwargs)
if image_input is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment