diff options
| author | pks <pks@pks.rocks> | 2025-12-01 11:24:14 +0100 |
|---|---|---|
| committer | pks <pks@pks.rocks> | 2025-12-01 11:24:14 +0100 |
| commit | 3645232a2e350b37a20deb67f88c654f15efb635 (patch) | |
| tree | 59ca989bb9c86282c3369a3ffbacacb0282a462c /finetuning.py | |
| parent | 54a77fc6145494644b98f2bab2320cbe3f9fcc6d (diff) | |
WIP
Diffstat (limited to 'finetuning.py')
| -rwxr-xr-x | finetuning.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/finetuning.py b/finetuning.py index 62f506c..a0ad1fe 100755 --- a/finetuning.py +++ b/finetuning.py @@ -80,7 +80,7 @@ def main(): 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: bnb_config = BitsAndBytesConfig( load_in_4bit=True, |