From ec2bf9757c53a03f925c80e3e73dbaa56ea6d3f6 Mon Sep 17 00:00:00 2001
From: chinainfant <38291328+chinainfant@users.noreply.github.com>
Date: Fri, 21 Jul 2023 15:28:11 +0800
Subject: [PATCH] =?UTF-8?q?Update=20args.py=20=E5=B0=86use-ptuning-v2?=
 =?UTF-8?q?=E7=9A=84=E5=91=BD=E5=90=8D=E8=A1=8C=E6=94=B9=E4=B8=BAaction=3D?=
 =?UTF-8?q?'store=5Ftrue'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/loader/args.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/loader/args.py b/models/loader/args.py
index e7df76b..cd3e78b 100644
--- a/models/loader/args.py
+++ b/models/loader/args.py
@@ -44,7 +44,7 @@ parser.add_argument('--no-remote-model', action='store_true', help='remote in th
 parser.add_argument('--model-name', type=str, default=LLM_MODEL, help='Name of the model to load by default.')
 parser.add_argument('--lora', type=str, help='Name of the LoRA to apply to the model by default.')
 parser.add_argument("--lora-dir", type=str, default=LORA_DIR, help="Path to directory with all the loras")
-parser.add_argument('--use-ptuning-v2',type=str,default=USE_PTUNING_V2,help="whether use ptuning-v2 checkpoint")
+parser.add_argument('--use-ptuning-v2',action='store_true',help="whether use ptuning-v2 checkpoint")
 parser.add_argument("--ptuning-dir",type=str,default=PTUNING_DIR,help="the dir of ptuning-v2 checkpoint")
 # Accelerate/transformers
 parser.add_argument('--load-in-8bit', action='store_true', default=LOAD_IN_8BIT,