From 05c8eba11c51ca928eee9c041de1a4192e590aec Mon Sep 17 00:00:00 2001
From: nianjiuhuiyi <64776403+nianjiuhuiyi@users.noreply.github.com>
Date: 星期五, 27 六月 2025 09:57:45 +0800
Subject: [PATCH] Fix: 修复c++后端服务因为空数组的异常退出，以及c++的http服务在收到Ctrl+C信号后无法正常退出 (#2571)
---
funasr/models/mossformer/mossformer_decoder.py | 5 +----
1 files changed, 1 insertions(+), 4 deletions(-)
diff --git a/funasr/models/mossformer/mossformer_decoder.py b/funasr/models/mossformer/mossformer_decoder.py
index e0189f7..56ad2ec 100644
--- a/funasr/models/mossformer/mossformer_decoder.py
+++ b/funasr/models/mossformer/mossformer_decoder.py
@@ -40,9 +40,7 @@
"""
if x.dim() not in [2, 3]:
- raise RuntimeError(
- "{} accept 3/4D tensor as input".format(self.__name__)
- )
+ raise RuntimeError("{} accept 3/4D tensor as input".format(self.__name__))
x = super().forward(x if x.dim() == 3 else torch.unsqueeze(x, 1))
if torch.squeeze(x).dim() == 1:
@@ -50,4 +48,3 @@
else:
x = torch.squeeze(x)
return x
-
--
Gitblit v1.9.1