嘉渊 2 лет назад
Родитель
Сommit
ba4aef18ed
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      egs/aishell/transformer/utils/gen_modelscope_configuration.py

+ 2 - 1
egs/aishell/transformer/utils/gen_modelscope_configuration.py

@@ -106,11 +106,12 @@ if __name__ == '__main__':
                                                                            args.dataset, args.vocab_size, args.tag),
         }
     }
+    pipeline = {"type": args.pipeline_type}
     json_dict = {
         "framework": "pytorch",
         "task": args.task,
         "model": model,
-        "pipeline": args.pipeline_type,
+        "pipeline": pipeline,
     }
 
     with open(os.path.join(args.output_dir, "configuration.json"), "w") as f: