From 59bc02b089f7a626fe67907dcfc695eae6883f82 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 14 六月 2024 13:59:49 +0800
Subject: [PATCH] decoding
---
funasr/models/sense_voice/model.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/models/sense_voice/model.py b/funasr/models/sense_voice/model.py
index 9be5abe..c77930d 100644
--- a/funasr/models/sense_voice/model.py
+++ b/funasr/models/sense_voice/model.py
@@ -1042,6 +1042,7 @@
self.length_normalized_loss = length_normalized_loss
self.beam_search = None
self.activation_checkpoint = kwargs.get("activation_checkpoint", False)
+ self.encoder_output_size = encoder_output_size
def forward(
self,
@@ -1451,6 +1452,7 @@
self.ctc = ctc
self.length_normalized_loss = length_normalized_loss
+ self.encoder_output_size = encoder_output_size
def forward(
self,
--
Gitblit v1.9.1