From e71546b06d0bddd30adf7edde9db7bee52b42b9f Mon Sep 17 00:00:00 2001
From: shixian <shixian@U-09RYG5WD-2244.local>
Date: 星期四, 05 十二月 2024 15:14:47 +0800
Subject: [PATCH] debug
---
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