From b3fcd42bf6e60372ed0f0d3e53798fcdd713fe08 Mon Sep 17 00:00:00 2001 From: Shi Xian <40013335+R1ckShi@users.noreply.github.com> Date: 星期三, 06 十二月 2023 11:14:57 +0800 Subject: [PATCH] Merge pull request #1144 from yeyupiaoling/main --- funasr/modules/streaming_utils/chunk_utilis.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/modules/streaming_utils/chunk_utilis.py b/funasr/modules/streaming_utils/chunk_utilis.py index ed8b31e..73dbaf4 100644 --- a/funasr/modules/streaming_utils/chunk_utilis.py +++ b/funasr/modules/streaming_utils/chunk_utilis.py @@ -61,7 +61,7 @@ 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) -- Gitblit v1.9.1