From eed5cbb9457f0842d9ea8f41cd88459a5dbc9e70 Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期二, 12 九月 2023 12:55:19 +0800
Subject: [PATCH] fix decoding_ind none bug
---
funasr/bin/asr_infer.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/asr_infer.py b/funasr/bin/asr_infer.py
index 7746821..79ab974 100644
--- a/funasr/bin/asr_infer.py
+++ b/funasr/bin/asr_infer.py
@@ -431,6 +431,8 @@
# b. Forward Encoder
if decoding_ind is None:
decoding_ind = self.decoding_ind
+ if self.decoding_ind is None:
+ decoding_ind = 0
enc, enc_len = self.asr_model.encode(**batch, ind=decoding_ind)
if isinstance(enc, tuple):
enc = enc[0]
--
Gitblit v1.9.1