Procházet zdrojové kódy

add paraformer online infer and finetune recipe

haoneng.lhn před 2 roky
rodič
revize
a8381b2b57

+ 1 - 0
egs_modelscope/asr/paraformer/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online/finetune.py

@@ -14,6 +14,7 @@ def modelscope_finetune(params):
     ds_dict = MsDataset.load(params.data_path)
     ds_dict = MsDataset.load(params.data_path)
     kwargs = dict(
     kwargs = dict(
         model=params.model,
         model=params.model,
+        model_revision='v1.0.6',
         data_dir=ds_dict,
         data_dir=ds_dict,
         dataset_type=params.dataset_type,
         dataset_type=params.dataset_type,
         work_dir=params.output_dir,
         work_dir=params.output_dir,

+ 1 - 1
funasr/bin/build_trainer.py

@@ -23,7 +23,7 @@ def parse_args(mode):
         from funasr.tasks.asr import ASRTask as ASRTask
         from funasr.tasks.asr import ASRTask as ASRTask
     elif mode == "paraformer":
     elif mode == "paraformer":
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask
-    elif mode == "paraformer_online":
+    elif mode == "paraformer_streaming":
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask
     elif mode == "paraformer_vad_punc":
     elif mode == "paraformer_vad_punc":
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask
         from funasr.tasks.asr import ASRTaskParaformer as ASRTask