fix fire_red_vad fsmn get cache bug

2026-04-16 21:30:20 +08:00
parent 961c6de3a5
commit baf58fc7ca
1 changed files with 7 additions and 2 deletions
@@ -86,8 +86,13 @@ impl FSMN {
        } else {
            residual.clone()
        };
-        let start = inputs.dim(D::Minus1)? - self.lookback_padding;
-        let new_cache = inputs.narrow(D::Minus1, start, self.lookback_padding)?;
+        let len = inputs.dim(D::Minus1)?;
+        let new_cache = if len > self.lookback_padding {
+            let start = len - self.lookback_padding;
+            inputs.narrow(D::Minus1, start, self.lookback_padding)?
+        } else {
+            inputs.clone()
+        };
        // conv1d_depthwise仅支持dilation=1的情况
        let lookback = if self.s1 == 1 {
            let inputs =