fix: add on_init_end before adding tc callback (#621)

YashasviChaurasia · web-flow · commit 6f2134d2667f · 2025-10-28T18:04:08.000+05:30
Signed-off-by: yashasvi &lt;yashasvi@ibm.com&gt;
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -513,6 +513,7 @@ def train(`
`513`	`513`	`tc_callback = TrainerControllerCallback(`
`514`	`514`	`trainer_controller_args.trainer_controller_config_file,`
`515`	`515`	`)`
	`516`	`+ tc_callback.on_init_end(trainer.args, trainer.state, trainer.control)`
`516`	`517`	`trainer.add_callback(tc_callback)`
`517`	`518`
`518`	`519`	`trainer.train(resume_from_checkpoint)`