Update ultravox_model.py

ultravox_model.py

@@ -201,12 +201,8 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
             for i, (audio, start, length) in enumerate(
                 zip(audio_embeds, audio_token_start_idx, audio_token_len)
             ):
-                try:
-                    inputs_embeds[i, start : start + length] = audio[:length]
-                except:
-                    print(f"{i=}, {audio.shape=}, {start=}, {length=}, {inputs_embeds[i].shape=}")
-                    print(f"{labels[i]=}")
-                    exit(1)
+                inputs_embeds[i, start : start + length] = audio
+
 
         lm_output = self.language_model.forward(
             inputs_embeds=inputs_embeds,