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

Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
Li Wang 2025-12-22 18:15:54 +08:00 committed by GitHub
parent c02a2705f9
commit 256a33ecb4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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: