Update train_lora.py

pranavps · Oct 27, 2023 · c73053c · c73053c
1 parent 2e69433
commit c73053c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fingpt/FinGPT_Benchmark/train_lora.py b/fingpt/FinGPT_Benchmark/train_lora.py
@@ -173,7 +173,7 @@ def main(args):
     parser.add_argument("--run_name", default='local-test', type=str)
     parser.add_argument("--dataset", required=True, type=str)
     parser.add_argument("--test_dataset", type=str)
-    parser.add_argument("--base_model", required=True, type=str, choices=['chatglm2', 'llama2','llama2-13b,'llama2-13b-nr','baichuan', 'falcon', 'internlm', 'qwen', 'mpt', 'bloom'])
+    parser.add_argument("--base_model", required=True, type=str, choices=['chatglm2', 'llama2', 'llama2-13b', 'llama2-13b-nr', 'baichuan', 'falcon', 'internlm', 'qwen', 'mpt', 'bloom'])
     parser.add_argument("--max_length", default=512, type=int)
     parser.add_argument("--batch_size", default=4, type=int, help="The train batch size per device")
     parser.add_argument("--learning_rate", default=1e-4, type=float, help="The learning rate")