[Model] Remove unused handle_oov_mm_token (#37321)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
@@ -185,14 +185,16 @@ def make_mock_model(hidden: int = 8):
|
||||
|
||||
# super().embed_input_ids → use SupportsMultiModal.embed_input_ids
|
||||
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(
|
||||
model,
|
||||
ids,
|
||||
mm_embs,
|
||||
is_multimodal=is_multimodal,
|
||||
handle_oov_mm_token=handle_oov_mm_token,
|
||||
)
|
||||
|
||||
# Bind embed_input_ids as the real method
|
||||
|
||||
@@ -793,7 +793,6 @@ class FireRedASR2ForConditionalGeneration(
|
||||
multimodal_embeddings: MultiModalEmbeddings | None = None,
|
||||
*,
|
||||
is_multimodal: torch.Tensor | None = None,
|
||||
handle_oov_mm_token: bool = False,
|
||||
) -> torch.Tensor:
|
||||
inputs_embeds = self.model.decoder.embed_input_ids(input_ids)
|
||||
|
||||
|
||||
@@ -514,7 +514,6 @@ class KimiAudioForConditionalGeneration(
|
||||
multimodal_embeddings: tuple[torch.Tensor, ...] | None = None,
|
||||
*,
|
||||
is_multimodal: torch.Tensor | None = None,
|
||||
handle_oov_mm_token: bool = False,
|
||||
) -> torch.Tensor:
|
||||
"""Embed input IDs and fuse with audio embeddings.
|
||||
|
||||
|
||||
Reference in New Issue
Block a user