AlexHung29629
/

test_mllama_v12

Feature Extraction

Model card Files Files and versions Community

AlexHung29629 commited on 28 days ago

Commit

424f6bf

•

1 Parent(s): 2f0402a

Update ultravox_model.py

Files changed (1) hide show

ultravox_model.py +1 -1

ultravox_model.py CHANGED Viewed

@@ -201,7 +201,7 @@ class UltravoxModel(transformers.LlamaPreTrainedModel):
             for i, (audio, start, length) in enumerate(
                 zip(audio_embeds, audio_token_start_idx, audio_token_len)
             ):
-                assert audio.shape[0] == length, f"{audio.shape=}, {start=}, {length=}, {inputs_embeds[i].shape=}"
                 inputs_embeds[i, start : start + length] = audio

             for i, (audio, start, length) in enumerate(
                 zip(audio_embeds, audio_token_start_idx, audio_token_len)
             ):
+                assert audio.shape[0] == length, f"{audio.shape=}, {start=}, {length=}, {inputs_embeds[i].shape=}, {audio_tower_output[i].shape=}, {audio_values.shape=}, {audio_len=}"
                 inputs_embeds[i, start : start + length] = audio