From 7fd45074ab47b2316c6a451981ad9f1f28cfead7 Mon Sep 17 00:00:00 2001 From: pks Date: Sun, 30 Nov 2025 22:19:59 +0100 Subject: WIP --- finetuning.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'finetuning.py') 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) -- cgit v1.2.3