From 7e5cbf5f5d388db887b0d3c71f3353ca9ddc10c4 Mon Sep 17 00:00:00 2001
From: yashasvi <yashasvi@ibm.com>
Date: Tue, 28 Oct 2025 17:45:46 +0530
Subject: [PATCH] fix: add on_init_end before adding tc callback

Signed-off-by: yashasvi <yashasvi@ibm.com>
---
 tuning/sft_trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
index c093a4d819..fd759229c7 100644
--- a/tuning/sft_trainer.py
+++ b/tuning/sft_trainer.py
@@ -513,6 +513,7 @@ def train(
         tc_callback = TrainerControllerCallback(
             trainer_controller_args.trainer_controller_config_file,
         )
+        tc_callback.on_init_end(trainer.args, trainer.state, trainer.control)
         trainer.add_callback(tc_callback)
 
     trainer.train(resume_from_checkpoint)