Merge pull request #932 from alibaba-damo-academy/dev_lhn

hnluo

2023-09-11 9fcb3cc06b4e324f0913d2f61b89becc2baeef1b

 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",
@@ -106,12 +106,13 @@
                                                                           args.dataset, args.vocab_size, args.tag),
        }
    }
    pipeline = {"type": args.pipeline_type}
    json_dict = {
        "model": model,
        "framework": "pytorch",
        "task": args.task,
        "pipeline": args.pipeline_type,
        "model": model,
        "pipeline": pipeline,
    }

    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)