Update README.md

README.md

@@ -66,13 +66,13 @@ from fengshen.pipelines.multiplechoice import UniMCPipelines
 total_parser = argparse.ArgumentParser("TASK NAME")
 total_parser = UniMCPipelines.piplines_args(total_parser)
 args = total_parser.parse_args()
-args.pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniMC-DeBERTa-v2-330M-Chinese'
+pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniMC-DeBERTa-v2-330M-Chinese'
 args.learning_rate=2e-5
 args.max_length=512
 args.max_epochs=3
 args.batchsize=8
 args.default_root_dir='./'
-model = UniMCPipelines(args)
+model = UniMCPipelines(args, pretrained_model_path)
 
 train_data = []
 dev_data = []