From de93c258d3ac6816c9956210d911cdd628fd18db Mon Sep 17 00:00:00 2001 From: pks Date: Sun, 30 Nov 2025 22:54:45 +0100 Subject: WIP --- finetuning.py | 1 + 1 file changed, 1 insertion(+) diff --git a/finetuning.py b/finetuning.py index f4543b7..853eb81 100755 --- a/finetuning.py +++ b/finetuning.py @@ -78,6 +78,7 @@ def main(): parser.add_argument("--lora-small", action="store_true") parser.add_argument("--max-grad-norm", default=1.0, type=float) parser.add_argument("--max-length", default=512, type=int) + parser.add_argument("--lora-config", choices=["S", "M", "L"], default="M") args = parser.parse_args() if args.bnb_4bit: -- cgit v1.2.3