egs/aishell/transformer/utils/gen_modelscope_configuration.py
@@ -106,12 +106,13 @@ 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: json.dump(json_dict, f) json.dump(json_dict, f, indent=4)
