From 759e8673ce497125da5855a173fd80f57bb071b3 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 10 Jun 2023 14:25:21 -0400
Subject: [PATCH] Update scripts/finetune.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
---
 scripts/finetune.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/scripts/finetune.py b/scripts/finetune.py
index 47aada411..cd9234334 100644
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -261,7 +261,6 @@ def train(
         model.save_pretrained(cfg.output_dir)
         return
 
-    model.train()
 
     trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)