egs/aishell/transformer/utils/gen_modelscope_configuration.py
@@ -81,7 +81,7 @@ "--nat", type=str, default="", help="vocab_size", help="nat", ) parser.add_argument( "--tag", @@ -107,11 +107,11 @@ } } json_dict = { "model": model, "framework": "pytorch", "task": args.task, "model": model, "pipeline": args.pipeline_type, } with open(os.path.join(args.output_dir, "configuration.json"), "w") as f: json.dump(json_dict, f) json.dump(json_dict, f, indent=4)
