From 161c9e6c509278768a2ddd9db53ba67287e951b7 Mon Sep 17 00:00:00 2001 From: hzg0601 Date: Fri, 28 Jul 2023 03:59:29 +0800 Subject: [PATCH] fix typo --- 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 02f4926..a62b4cf 100644 --- a/models/loader/args.py +++ b/models/loader/args.py @@ -45,7 +45,7 @@ parser.add_argument('--model-name', type=str, default=LLM_MODEL, help='Name of t parser.add_argument("--use-lora",type=bool,default=USE_LORA,help="use lora or not") parser.add_argument('--lora', type=str, default=LORA_NAME,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',action=USE_PTUNING_V2,help="whether use ptuning-v2 checkpoint") +parser.add_argument('--use-ptuning-v2',default=USE_PTUNING_V2,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,