openbmb
/

MiniCPM-o-2_6

Model card Files Files and versions

Update modeling_minicpmo.py

#32

by BUAADreamer - opened Feb 10, 2025

base: refs/heads/main

←

from: refs/pr/32

Discussion Files changed

Files changed (1) hide show

modeling_minicpmo.py +2 -0

modeling_minicpmo.py CHANGED Viewed

@@ -584,6 +584,8 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
                 if self.config.chunk_input:
                     for i in range(bs):
                         audio_embs = torch.cat(audio_embeddings[i], dim=0).to(
                             device=input_embeddings.device, dtype=input_embeddings.dtype
                         )

                 if self.config.chunk_input:
                     for i in range(bs):
+                        if not audio_embeddings[i]:
+                            continue
                         audio_embs = torch.cat(audio_embeddings[i], dim=0).to(
                             device=input_embeddings.device, dtype=input_embeddings.dtype
                         )