From f43da18b5e097515c12438c70fc7918c29e5d0c8 Mon Sep 17 00:00:00 2001
From: lingji-yidong <75744976+lingji-yidong@users.noreply.github.com>
Date: 星期一, 19 八月 2024 13:36:59 +0800
Subject: [PATCH] fix start timestamp in sentence_info (#2024)
---
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