From 5e06b2a94fd78d8b4fa4338ea311bd3699497b93 Mon Sep 17 00:00:00 2001 From: Saeed Dehqan <31902891+saeeddhqan@users.noreply.github.com> Date: Wed, 20 Nov 2024 01:29:37 -0500 Subject: [PATCH] Update ultravox_model_test.py --- ultravox/model/ultravox_model_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultravox/model/ultravox_model_test.py b/ultravox/model/ultravox_model_test.py index 51c467ad..a0b95b20 100644 --- a/ultravox/model/ultravox_model_test.py +++ b/ultravox/model/ultravox_model_test.py @@ -31,7 +31,7 @@ def test_init_latency_mask_valid(encoder): assert encoder.audio_streaming_mask.shape[1] == 1 mask = encoder.audio_streaming_mask[0, 0] - # 50x60=3000 + # 100*30=3000 source_mask = ( torch.tril(torch.ones(30, 30), diagonal=0) .repeat_interleave(block_size, dim=0)