From dd3a3054ddb67989bda8b414e54be99f6a13d08d Mon Sep 17 00:00:00 2001 From: hnluo Date: Sun, 5 Feb 2023 11:59:39 +0800 Subject: [PATCH] add sampling rate parameter --- funasr/bin/asr_inference_paraformer_vad_punc.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/funasr/bin/asr_inference_paraformer_vad_punc.py b/funasr/bin/asr_inference_paraformer_vad_punc.py index 10c1da64f..27026177f 100644 --- a/funasr/bin/asr_inference_paraformer_vad_punc.py +++ b/funasr/bin/asr_inference_paraformer_vad_punc.py @@ -548,6 +548,7 @@ def inference_modelscope( def _forward(data_path_and_name_and_type, raw_inputs: Union[np.ndarray, torch.Tensor] = None, output_dir_v2: Optional[str] = None, + fs: dict = None, param_dict: dict = None, ): # 3. Build data-iterator @@ -558,6 +559,7 @@ def inference_modelscope( loader = ASRTask.build_streaming_iterator( data_path_and_name_and_type, dtype=dtype, + fs=fs, batch_size=1, key_file=key_file, num_workers=num_workers,