Update finetune.py

finetune.py

@@ -119,7 +119,7 @@ training_args = TrainingArguments(
     save_total_limit=2,
     load_best_model_at_end=True,
     metric_for_best_model="eval_loss",
-    use_peft=True,
+    #use_peft=True,
 )
 
 # ============================================
@@ -130,7 +130,8 @@ trainer = SFTTrainer(
     train_dataset=tokenized_datasets["train"],
     eval_dataset=tokenized_datasets["validation"],
     peft_config=lora_config,
-    args=training_args
+    args=training_args,
+    use_peft=True,
 )
 
 # 9. Fine-tuning