add len(audio_embeddings) > 0 judge

#17
Files changed (1) hide show
  1. 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
- input_embeddings = input_embeddings + audio_embeddings[0].mean() * 0
 
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