From cab7bdaa16ce1022c089e4ee46bbc68c67faf237 Mon Sep 17 00:00:00 2001 From: hnluo Date: Sun, 5 Feb 2023 12:00:40 +0800 Subject: [PATCH] add sampling rate parameter --- funasr/bin/asr_inference_uniasr.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/funasr/bin/asr_inference_uniasr.py b/funasr/bin/asr_inference_uniasr.py index a4bdcc1a4..8a9905830 100644 --- a/funasr/bin/asr_inference_uniasr.py +++ b/funasr/bin/asr_inference_uniasr.py @@ -575,6 +575,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 @@ -585,6 +586,7 @@ def inference_modelscope( loader = ASRTask.build_streaming_iterator( data_path_and_name_and_type, dtype=dtype, + fs=fs, batch_size=batch_size, key_file=key_file, num_workers=num_workers,