hnluo
2023-02-05 cab7bdaa16ce1022c089e4ee46bbc68c67faf237
add sampling rate parameter
1个文件已修改
2 ■■■■■ 已修改文件
funasr/bin/asr_inference_uniasr.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/asr_inference_uniasr.py
@@ -575,6 +575,7 @@
    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 @@
        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,