Update ultravox_model.py

ultravox_model.py

@@ -204,8 +204,8 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
                 try:
                     inputs_embeds[i, start : start + length] = audio[:length]
                 except:
-                    print(f"{audio.shape=}, {start=}, {length=}, {audio.shape=}")
-                    print(f"{labels=}")
+                    print(f"{i=}, {audio.shape=}, {start=}, {length=}, {inputs_embeds[i].shape=}")
+                    print(f"{labels[i]=}")
                     exit(1)
 
         lm_output = self.language_model.forward(