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/transformer/positionwise_feed_forward.py | 21 ++++++++++----------- 1 files changed, 10 insertions(+), 11 deletions(-) diff --git a/funasr/models/transformer/positionwise_feed_forward.py b/funasr/models/transformer/positionwise_feed_forward.py index 081ff5b..7cfa5f9 100644 --- a/funasr/models/transformer/positionwise_feed_forward.py +++ b/funasr/models/transformer/positionwise_feed_forward.py @@ -35,15 +35,14 @@ class PositionwiseFeedForwardDecoderSANMExport(torch.nn.Module): - def __init__(self, model): - super().__init__() - self.w_1 = model.w_1 - self.w_2 = model.w_2 - self.activation = model.activation - self.norm = model.norm - - def forward(self, x): - x = self.activation(self.w_1(x)) - x = self.w_2(self.norm(x)) - return x + def __init__(self, model): + super().__init__() + self.w_1 = model.w_1 + self.w_2 = model.w_2 + self.activation = model.activation + self.norm = model.norm + def forward(self, x): + x = self.activation(self.w_1(x)) + x = self.w_2(self.norm(x)) + return x -- Gitblit v1.9.1