From 0454015357a69bad81002b24b3806aa98ae2ef14 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 30 四月 2024 00:28:14 +0800
Subject: [PATCH] batch
---
funasr/models/sense_voice/decoder.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/models/sense_voice/decoder.py b/funasr/models/sense_voice/decoder.py
index 4986d50..f5b8825 100644
--- a/funasr/models/sense_voice/decoder.py
+++ b/funasr/models/sense_voice/decoder.py
@@ -365,7 +365,7 @@
self.pad_fn = nn.ConstantPad1d((left_padding, right_padding), 0.0)
self.kernel_size = kernel_size
- def forward(self, inputs, mask, cache=None, mask_shfit_chunk=None):
+ def forward(self, inputs, mask, cache=None, mask_shfit_chunk=None, **kwargs):
"""
:param x: (#batch, time1, size).
:param mask: Mask tensor (#batch, 1, time)
--
Gitblit v1.9.1