From a3a1c55c4c674768e6cfef1dc9c10d2e1c3335c7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 25 十二月 2024 17:27:10 +0800
Subject: [PATCH] v1.2.2
---
tests/test_lm_pipeline.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/tests/test_lm_pipeline.py b/tests/test_lm_pipeline.py
index 3a5ec57..f67c9a5 100644
--- a/tests/test_lm_pipeline.py
+++ b/tests/test_lm_pipeline.py
@@ -6,20 +6,22 @@
logger = get_logger()
+
class TestTransformerInferencePipelines(unittest.TestCase):
def test_funasr_path(self):
import funasr
import os
+
logger.info("run_dir:{0} ; funasr_path: {1}".format(os.getcwd(), funasr.__file__))
def test_inference_pipeline(self):
inference_pipeline = pipeline(
task=Tasks.language_score_prediction,
- model='damo/speech_transformer_lm_zh-cn-common-vocab8404-pytorch',
+ model="damo/speech_transformer_lm_zh-cn-common-vocab8404-pytorch",
)
rec_result = inference_pipeline(text_in="hello 澶� 瀹� 濂� 鍛�")
logger.info("lm inference result: {0}".format(rec_result))
-if __name__ == '__main__':
+if __name__ == "__main__":
unittest.main()
--
Gitblit v1.9.1