From 8b0fb74bded1f8a162e6c0e94c3522be6216ea03 Mon Sep 17 00:00:00 2001
From: chengligen <101448376+chengligen@users.noreply.github.com>
Date: 星期一, 26 五月 2025 14:11:33 +0800
Subject: [PATCH] feat: add 'words' key aligned with timestamps in sensevoice model output (#2531)
---
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