From 9c5f00794e7c474d1079ff78e5c69e8c53a30e6e Mon Sep 17 00:00:00 2001 From: dudulu <solacowa@gmail.com> Date: 星期一, 05 二月 2024 10:10:52 +0800 Subject: [PATCH] 修复chunk_size可能传入str或list结构 (#1357) --- examples/industrial_data_pretraining/seaco_paraformer/demo.py | 3 ++- 1 files changed, 2 insertions(+), 1 deletions(-) diff --git a/examples/industrial_data_pretraining/seaco_paraformer/demo.py b/examples/industrial_data_pretraining/seaco_paraformer/demo.py index e9e226d..85d989e 100644 --- a/examples/industrial_data_pretraining/seaco_paraformer/demo.py +++ b/examples/industrial_data_pretraining/seaco_paraformer/demo.py @@ -25,6 +25,7 @@ # example2 import torchaudio +import os wav_file = os.path.join(model.model_path, "example/asr_example.wav") input_tensor, sample_rate = torchaudio.load(wav_file) input_tensor = input_tensor.mean(0) @@ -33,7 +34,7 @@ # example3 import soundfile -import os + wav_file = os.path.join(model.model_path, "example/asr_example.wav") speech, sample_rate = soundfile.read(wav_file) res = model.generate(input=[speech], batch_size_s=300, is_final=True) -- Gitblit v1.9.1