Fix block size (#54)

EleutherAI · Aug 30, 2024 · c137ce7 · c137ce7
1 parent 774e342
commit c137ce7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/amt/inference/transcribe.py b/amt/inference/transcribe.py
@@ -35,7 +35,7 @@
 torch._inductor.config.fx_graph_cache = True
 
 MAX_SEQ_LEN = 4096
-MAX_BLOCK_LEN = 2048
+MAX_BLOCK_LEN = 4096
 LEN_MS = 30000
 STRIDE_FACTOR = 3
 CHUNK_LEN_MS = LEN_MS // STRIDE_FACTOR