From 1f6370e350b16dd18499ce2ae7f32ad9216a62f8 Mon Sep 17 00:00:00 2001 From: "shixian.shi" Date: Thu, 20 Apr 2023 11:23:08 +0800 Subject: [PATCH] bug fix --- funasr/bin/tp_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/bin/tp_inference.py b/funasr/bin/tp_inference.py index 191bbf325..df029fdcc 100644 --- a/funasr/bin/tp_inference.py +++ b/funasr/bin/tp_inference.py @@ -54,7 +54,7 @@ class SpeechText2Timestamp: assert check_argument_types() # 1. Build ASR model tp_model, tp_train_args = ASRTask.build_model_from_file( - timestamp_infer_config, timestamp_model_file, device + timestamp_infer_config, timestamp_model_file, device=device ) if 'cuda' in device: tp_model = tp_model.cuda() # force model to cuda