yuzaa commited on
Commit
1bbb766
·
verified ·
1 Parent(s): 4e1cff2

fix audio batch inference

Browse files
Files changed (1) hide show
  1. modeling_minicpmo.py +1 -1
modeling_minicpmo.py CHANGED
@@ -589,7 +589,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
589
  audio_start_pos = 0
590
  for bound in audio_bounds[i]:
591
  audio_len = bound[1] - bound[0]
592
- input_embeddings[0, bound[0] : bound[1]] = audio_embs[
593
  audio_start_pos : audio_start_pos + audio_len, :
594
  ]
595
  audio_start_pos += audio_len
 
589
  audio_start_pos = 0
590
  for bound in audio_bounds[i]:
591
  audio_len = bound[1] - bound[0]
592
+ input_embeddings[i, bound[0] : bound[1]] = audio_embs[
593
  audio_start_pos : audio_start_pos + audio_len, :
594
  ]
595
  audio_start_pos += audio_len