From ca98012cd225706e921df249090c6cca48b491d1 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期五, 10 三月 2023 10:50:28 +0800
Subject: [PATCH] update tp_inference
---
funasr/bin/tp_inference.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/tp_inference.py b/funasr/bin/tp_inference.py
index 3bd3324..a2472cf 100644
--- a/funasr/bin/tp_inference.py
+++ b/funasr/bin/tp_inference.py
@@ -263,7 +263,7 @@
preprocessor = LMPreprocessor(
train=False,
token_type=speechtext2timestamp.tp_train_args.token_type,
- token_list=speechtext2timestamp.tp_train_args,
+ token_list=speechtext2timestamp.tp_train_args.token_list,
bpemodel=None,
text_cleaner=None,
g2p_type=None,
@@ -293,7 +293,7 @@
batch_size=batch_size,
key_file=key_file,
num_workers=num_workers,
- preprocess_fn=LMPreprocessor,
+ preprocess_fn=preprocessor,
collate_fn=ASRTask.build_collate_fn(speechtext2timestamp.tp_train_args, False),
allow_variable_data_keys=allow_variable_data_keys,
inference=True,
--
Gitblit v1.9.1