diff --git a/egs/aishell/paraformer/run.sh b/egs/aishell/paraformer/run.sh index e9174042a..b7e3a827e 100755 --- a/egs/aishell/paraformer/run.sh +++ b/egs/aishell/paraformer/run.sh @@ -219,7 +219,7 @@ if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then --am_model_name $inference_asr_model \ --mode paraformer \ --model_name paraformer \ - --model $model_dir \ + --dataset aishell \ --output_dir $exp_dir/exp/$model_dir \ --vocab_size $vocab_size \ --nat _nat \ diff --git a/egs/aishell/transformer/utils/gen_modelscope_configuration.py b/egs/aishell/transformer/utils/gen_modelscope_configuration.py index 84146475a..7962f2336 100644 --- a/egs/aishell/transformer/utils/gen_modelscope_configuration.py +++ b/egs/aishell/transformer/utils/gen_modelscope_configuration.py @@ -81,7 +81,7 @@ if __name__ == '__main__': "--nat", type=str, default="", - help="vocab_size", + help="nat", ) parser.add_argument( "--tag", @@ -107,9 +107,9 @@ if __name__ == '__main__': } } json_dict = { - "model": model, "framework": "pytorch", "task": args.task, + "model": model, "pipeline": args.pipeline_type, }