diff options
| -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) |