add sampling rate parameter

This commit is contained in:
hnluo 2023-02-05 12:00:40 +08:00 committed by GitHub
parent dd3a3054dd
commit cab7bdaa16
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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,