From f3330efc0abc868dd5680832025bcf11e252f280 Mon Sep 17 00:00:00 2001 From: hnluo Date: Wed, 25 Oct 2023 11:00:57 +0800 Subject: [PATCH] fix chunk size select bug --- funasr/modules/streaming_utils/chunk_utilis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/modules/streaming_utils/chunk_utilis.py b/funasr/modules/streaming_utils/chunk_utilis.py index ed8b31eb2..73dbaf49e 100644 --- a/funasr/modules/streaming_utils/chunk_utilis.py +++ b/funasr/modules/streaming_utils/chunk_utilis.py @@ -61,7 +61,7 @@ class overlap_chunk(): chunk_num = len(self.chunk_size) ind = 0 if training and chunk_num > 1: - ind = torch.randint(0, chunk_num-1, ()).cpu().item() + ind = torch.randint(0, chunk_num, ()).cpu().item() if not training and decoding_ind is not None: ind = int(decoding_ind)