alex-ht commited on
Commit
fc81040
·
1 Parent(s): b044816
Files changed (1) hide show
  1. ultravox_model.py +0 -2
ultravox_model.py CHANGED
@@ -201,8 +201,6 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
201
  for i, (audio, start, length) in enumerate(
202
  zip(audio_embeds, audio_token_start_idx, audio_token_len)
203
  ):
204
- length = min(length, audio.shape[0])
205
- length = min(inputs_embeds.shape[1] - start, length)
206
  inputs_embeds[i, start : start + length] = audio[:length]
207
 
208
  lm_output = self.language_model.forward(
 
201
  for i, (audio, start, length) in enumerate(
202
  zip(audio_embeds, audio_token_start_idx, audio_token_len)
203
  ):
 
 
204
  inputs_embeds[i, start : start + length] = audio[:length]
205
 
206
  lm_output = self.language_model.forward(