Update scripts/finetune.py

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
2023-06-10 14:25:21 -04:00
parent 0c6f928601
commit 759e8673ce
1 changed files with 0 additions and 1 deletions
--- 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)