speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update param list

Browse Source
This commit is contained in:
jmwang66 2023-02-03 11:03:42 +08:00
parent 65b35ca423
commit 67b9af91d2
4 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/asr_inference.py
View File

@ -483,6 +483,7 @@ def inference_modelscope(
ngram_weight: float = 0.9, ngram_weight: float = 0.9,
nbest: int = 1, nbest: int = 1,
num_workers: int = 1, num_workers: int = 1,
param_dict: dict = None,
**kwargs, **kwargs,
): ):
assert check_argument_types() assert check_argument_types()
@ -533,6 +534,7 @@ def inference_modelscope(
def _forward(data_path_and_name_and_type, def _forward(data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None, raw_inputs: Union[np.ndarray, torch.Tensor] = None,
output_dir_v2: Optional[str] = None, output_dir_v2: Optional[str] = None,
param_dict: dict = None,
): ):
# 3. Build data-iterator # 3. Build data-iterator
if data_path_and_name_and_type is None and raw_inputs is not None: if data_path_and_name_and_type is None and raw_inputs is not None:

2
funasr/bin/asr_inference_paraformer.py
View File

@ -527,6 +527,7 @@ def inference_modelscope(
nbest: int = 1, nbest: int = 1,
num_workers: int = 1, num_workers: int = 1,
output_dir: Optional[str] = None, output_dir: Optional[str] = None,
param_dict: dict = None,
**kwargs, **kwargs,
): ):
assert check_argument_types() assert check_argument_types()
@ -576,6 +577,7 @@ def inference_modelscope(
data_path_and_name_and_type, data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None, raw_inputs: Union[np.ndarray, torch.Tensor] = None,
output_dir_v2: Optional[str] = None, output_dir_v2: Optional[str] = None,
param_dict: dict = None,
): ):
# 3. Build data-iterator # 3. Build data-iterator
if data_path_and_name_and_type is None and raw_inputs is not None: if data_path_and_name_and_type is None and raw_inputs is not None:

2
funasr/bin/asr_inference_paraformer_vad_punc.py
View File

@ -479,6 +479,7 @@ def inference_modelscope(
punc_infer_config: Optional[str] = None, punc_infer_config: Optional[str] = None,
punc_model_file: Optional[str] = None, punc_model_file: Optional[str] = None,
outputs_dict: Optional[bool] = True, outputs_dict: Optional[bool] = True,
param_dict: dict = None,
**kwargs, **kwargs,
): ):
assert check_argument_types() assert check_argument_types()
@ -544,6 +545,7 @@ def inference_modelscope(
def _forward(data_path_and_name_and_type, def _forward(data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None, raw_inputs: Union[np.ndarray, torch.Tensor] = None,
output_dir_v2: Optional[str] = None, output_dir_v2: Optional[str] = None,
param_dict: dict = None,
): ):
# 3. Build data-iterator # 3. Build data-iterator
if data_path_and_name_and_type is None and raw_inputs is not None: if data_path_and_name_and_type is None and raw_inputs is not None:

2
funasr/bin/asr_inference_uniasr.py
View File

@ -521,6 +521,7 @@ def inference_modelscope(
token_num_relax: int = 1, token_num_relax: int = 1,
decoding_ind: int = 0, decoding_ind: int = 0,
decoding_mode: str = "model1", decoding_mode: str = "model1",
param_dict: dict = None,
**kwargs, **kwargs,
): ):
assert check_argument_types() assert check_argument_types()
@ -574,6 +575,7 @@ def inference_modelscope(
def _forward(data_path_and_name_and_type, def _forward(data_path_and_name_and_type,
raw_inputs: Union[np.ndarray, torch.Tensor] = None, raw_inputs: Union[np.ndarray, torch.Tensor] = None,
output_dir_v2: Optional[str] = None, output_dir_v2: Optional[str] = None,
param_dict: dict = None,
): ):
# 3. Build data-iterator # 3. Build data-iterator
if data_path_and_name_and_type is None and raw_inputs is not None: if data_path_and_name_and_type is None and raw_inputs is not None: