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