speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update repo

Browse Source
This commit is contained in:
嘉渊 2023-05-25 10:42:59 +08:00
parent 8e9b79cae3
commit ba4aef18ed
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/aishell/transformer/utils/gen_modelscope_configuration.py
View File

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