diff options
| author | pks <pks@pks.rocks> | 2025-11-30 22:19:59 +0100 |
|---|---|---|
| committer | pks <pks@pks.rocks> | 2025-11-30 22:19:59 +0100 |
| commit | 7fd45074ab47b2316c6a451981ad9f1f28cfead7 (patch) | |
| tree | 4e62824e505b93d4374df6eedd75dc76a444cf64 /finetuning.py | |
| parent | cff0a680751f92a0bc49a212dc03906e67d35894 (diff) | |
WIP
Diffstat (limited to 'finetuning.py')
| -rwxr-xr-x | finetuning.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/finetuning.py b/finetuning.py index b1cb7bd..b0cf1cb 100755 --- a/finetuning.py +++ b/finetuning.py @@ -86,8 +86,8 @@ def main(): bnb_4bit_compute_dtype=torch.bfloat16, bnb_4bit_quant_storage=torch.bfloat16, ) - else: - bnb_config = None + else: + bnb_config = None processor = AutoProcessor.from_pretrained(args.model, use_fast=True) |