add len(audio_embeddings) > 0 judge
#17
by
BUAADreamer
- opened
- modeling_minicpmo.py +2 -1
modeling_minicpmo.py
CHANGED
@@ -598,7 +598,8 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
|
|
598 |
elif self.training:
|
599 |
for i in range(bs):
|
600 |
# dummy audio_embeddings
|
601 |
-
|
|
|
602 |
|
603 |
return input_embeddings
|
604 |
|
|
|
598 |
elif self.training:
|
599 |
for i in range(bs):
|
600 |
# dummy audio_embeddings
|
601 |
+
if len(audio_embeddings)>0:
|
602 |
+
input_embeddings = input_embeddings + audio_embeddings[0].mean() * 0
|
603 |
|
604 |
return input_embeddings
|
605 |
|