From cc77bab5267b1e72a2d0333269da0637389761d5 Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue, 9 May 2023 00:41:19 +0900
Subject: [PATCH] Add callbacks to Trainer

---
 src/axolotl/utils/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py
index ce3e65ca4..baa0ce626 100644
--- a/src/axolotl/utils/trainer.py
+++ b/src/axolotl/utils/trainer.py
@@ -204,6 +204,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
             return_tensors="pt",
             **data_collator_kwargs,
         ),
+        callbacks=callbacks,
         **trainer_kwargs,
     )