From abb33d6b2097e5b0643326bc1b376a63cdc2f967 Mon Sep 17 00:00:00 2001 From: zhifu gao <zhifu.gzf@alibaba-inc.com> Date: 星期一, 24 六月 2024 17:06:21 +0800 Subject: [PATCH] Dev gzf deepspeed (#1844) --- runtime/python/libtorch/demo_paraformer.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/runtime/python/libtorch/demo_paraformer.py b/runtime/python/libtorch/demo_paraformer.py index 9c3a082..62355e2 100644 --- a/runtime/python/libtorch/demo_paraformer.py +++ b/runtime/python/libtorch/demo_paraformer.py @@ -1,5 +1,5 @@ from pathlib import Path -from funasr_torch import Paraformer +from funasr_torch.paraformer_bin import Paraformer model_dir = "iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" model = Paraformer(model_dir, batch_size=1) # cpu -- Gitblit v1.9.1