Update ultravox_model_test.py

fixie-ai · Nov 20, 2024 · 5e06b2a · 5e06b2a
1 parent 93164a8
commit 5e06b2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)