hnluo 3 лет назад
Родитель
Сommit
dd3a3054dd
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      funasr/bin/asr_inference_paraformer_vad_punc.py

+ 2 - 0
funasr/bin/asr_inference_paraformer_vad_punc.py

@@ -548,6 +548,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,
+                 fs: dict = None,
                  param_dict: dict = None,
                  param_dict: dict = None,
                  ):
                  ):
         # 3. Build data-iterator
         # 3. Build data-iterator
@@ -558,6 +559,7 @@ def inference_modelscope(
         loader = ASRTask.build_streaming_iterator(
         loader = ASRTask.build_streaming_iterator(
             data_path_and_name_and_type,
             data_path_and_name_and_type,
             dtype=dtype,
             dtype=dtype,
+            fs=fs,
             batch_size=1,
             batch_size=1,
             key_file=key_file,
             key_file=key_file,
             num_workers=num_workers,
             num_workers=num_workers,