From 77045e7bb78d4b8a82f96130f9d84e356a32d5c5 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期二, 09 五月 2023 11:16:07 +0800
Subject: [PATCH] rnnt bug fix
---
funasr/tasks/asr.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 87db05c..a64b9e7 100644
--- a/funasr/tasks/asr.py
+++ b/funasr/tasks/asr.py
@@ -1682,7 +1682,7 @@
# 7. Build model
- if encoder.unified_model_training:
+ if hasattr(encoder, 'unified_model_training') and encoder.unified_model_training:
model = UnifiedTransducerModel(
vocab_size=vocab_size,
token_list=token_list,
--
Gitblit v1.9.1