From c137ce770d8b8355f68a90d3c0c5cf4a527d8167 Mon Sep 17 00:00:00 2001 From: Louis Date: Fri, 30 Aug 2024 15:37:25 +0100 Subject: [PATCH] Fix block size (#54) --- amt/inference/transcribe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/amt/inference/transcribe.py b/amt/inference/transcribe.py index 057649d..996e469 100644 --- 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