This commit is contained in:
游雁 2023-02-14 15:35:47 +08:00
parent baa0143b4f
commit 560dafa8ad
2 changed files with 3 additions and 0 deletions

Binary file not shown.

Before

Width:  |  Height:  |  Size: 184 KiB

View File

@ -470,6 +470,7 @@ def inference_modelscope(
ngram_weight: float = 0.9,
nbest: int = 1,
num_workers: int = 1,
param_dict: dict = None,
**kwargs,
):
assert check_argument_types()
@ -520,6 +521,8 @@ 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
if data_path_and_name_and_type is None and raw_inputs is not None: