Unverified Commit 51f0acda authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Model] Remove unused `handle_oov_mm_token` (#37321)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent fa75204b
...@@ -185,14 +185,16 @@ def make_mock_model(hidden: int = 8): ...@@ -185,14 +185,16 @@ def make_mock_model(hidden: int = 8):
# super().embed_input_ids → use SupportsMultiModal.embed_input_ids # super().embed_input_ids → use SupportsMultiModal.embed_input_ids
def fake_super_embed( def fake_super_embed(
ids, mm_embs=None, *, is_multimodal=None, handle_oov_mm_token=False ids,
mm_embs=None,
*,
is_multimodal=None,
): ):
return SupportsMultiModal.embed_input_ids( return SupportsMultiModal.embed_input_ids(
model, model,
ids, ids,
mm_embs, mm_embs,
is_multimodal=is_multimodal, is_multimodal=is_multimodal,
handle_oov_mm_token=handle_oov_mm_token,
) )
# Bind embed_input_ids as the real method # Bind embed_input_ids as the real method
......
...@@ -793,7 +793,6 @@ class FireRedASR2ForConditionalGeneration( ...@@ -793,7 +793,6 @@ class FireRedASR2ForConditionalGeneration(
multimodal_embeddings: MultiModalEmbeddings | None = None, multimodal_embeddings: MultiModalEmbeddings | None = None,
*, *,
is_multimodal: torch.Tensor | None = None, is_multimodal: torch.Tensor | None = None,
handle_oov_mm_token: bool = False,
) -> torch.Tensor: ) -> torch.Tensor:
inputs_embeds = self.model.decoder.embed_input_ids(input_ids) inputs_embeds = self.model.decoder.embed_input_ids(input_ids)
......
...@@ -514,7 +514,6 @@ class KimiAudioForConditionalGeneration( ...@@ -514,7 +514,6 @@ class KimiAudioForConditionalGeneration(
multimodal_embeddings: tuple[torch.Tensor, ...] | None = None, multimodal_embeddings: tuple[torch.Tensor, ...] | None = None,
*, *,
is_multimodal: torch.Tensor | None = None, is_multimodal: torch.Tensor | None = None,
handle_oov_mm_token: bool = False,
) -> torch.Tensor: ) -> torch.Tensor:
"""Embed input IDs and fuse with audio embeddings. """Embed input IDs and fuse with audio embeddings.
......
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