嘉渊
2023-05-31 af3f3dd5fd82823eaf64c0523b8c1b0fc79a8d8e
egs/aishell/transformer/utils/gen_modelscope_configuration.py
@@ -106,11 +106,12 @@
                                                                           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: