嘉渊 2 lat temu
rodzic
commit
ba4aef18ed

+ 2 - 1
egs/aishell/transformer/utils/gen_modelscope_configuration.py

@@ -106,11 +106,12 @@ if __name__ == '__main__':
                                                                            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: