From fe6744fbd940a2032607d1fcdb2269a02cfc3dfc Mon Sep 17 00:00:00 2001 From: Jzz1943 Date: Sat, 10 Jan 2026 23:10:04 +0800 Subject: [PATCH] fix(fsmn): specify squeeze dim to make ONNX Squeeze axes explicit for TensorRT (#1576) (#1577) --- modelscope/models/audio/separation/m2/fsmn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/models/audio/separation/m2/fsmn.py b/modelscope/models/audio/separation/m2/fsmn.py index 97400888..07fb0f8b 100644 --- a/modelscope/models/audio/separation/m2/fsmn.py +++ b/modelscope/models/audio/separation/m2/fsmn.py @@ -141,4 +141,4 @@ class UniDeepFsmnDilated(nn.Module): out = self.conv(x_per) out1 = out.permute(0, 3, 2, 1) - return input + out1.squeeze() + return input + out1.squeeze(dim=1)