From 6fe10a8dbfbab2bdcc28a411f9c5da85a4a8d002 Mon Sep 17 00:00:00 2001
From: hohaiuhsx <hohaiuhsx@gmail.com>
Date: 星期一, 10 三月 2025 23:16:22 +0800
Subject: [PATCH] 修复 当选用SenseVoice模型处理长音频(同时开启vad和output_timestamp)时的异常 (#2413)

---
 funasr/frontends/utils/mask_estimator.py |    4 +---
 1 files changed, 1 insertions(+), 3 deletions(-)

diff --git a/funasr/frontends/utils/mask_estimator.py b/funasr/frontends/utils/mask_estimator.py
index 002e474..29a21d9 100644
--- a/funasr/frontends/utils/mask_estimator.py
+++ b/funasr/frontends/utils/mask_estimator.py
@@ -23,9 +23,7 @@
 
         self.type = type
         self.nmask = nmask
-        self.linears = torch.nn.ModuleList(
-            [torch.nn.Linear(projs, idim) for _ in range(nmask)]
-        )
+        self.linears = torch.nn.ModuleList([torch.nn.Linear(projs, idim) for _ in range(nmask)])
 
     def forward(
         self, xs: ComplexTensor, ilens: torch.LongTensor

--
Gitblit v1.9.1