update

main.py

@@ -18,8 +18,7 @@ def get_args():
     parser.add_argument("--valid_subset", default="valid.jsonl", type=str)
     parser.add_argument(
         "--pretrained_model_name_or_path",
-        # default="YeungNLP/firefly-chatglm2-6b",
-        default=(project_path / "trained_models/firefly_chatglm2_6b_intent").as_posix(),
+        default=(project_path / "trained_models/qwen_7b_chinese_modern_poetry").as_posix(),
         type=str
     )
     parser.add_argument("--output_file", default="result.xlsx", type=str)