update

fixie-ai · Nov 7, 2024 · 79e3c71 · 79e3c71
1 parent 80b88cb
commit 79e3c71
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ultravox/model/data_processing_test.py b/ultravox/model/data_processing_test.py
@@ -31,6 +31,7 @@ def fake_process(text, audio, return_tensors="pt", sampling_rate=16000):
         "audio_values": torch.tensor([[[0.1, 0.2, 0.3]]]),
         "audio_token_start_idx": torch.tensor([1]),
         "audio_token_len": torch.tensor([2]),
+        "audio_len": torch.tensor([10]),
     }