Skip to content

Commit 110c26e

Browse files
fix: concatenate the trainer_callbacks and additional_callbacks in sft_trainer:train()
Signed-off-by: Vassilis Vassiliadis <vassilis.vassiliadis@ibm.com>
1 parent 10bc5b5 commit 110c26e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tuning/sft_trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def train(
150150

151151
# Add any extra callback if passed by users
152152
if additional_callbacks is not None:
153-
trainer_callbacks.append(additional_callbacks)
153+
trainer_callbacks.extend(additional_callbacks)
154154

155155
framework = AccelerationFrameworkConfig.from_dataclasses(
156156
quantized_lora_config, fusedops_kernels_config

0 commit comments

Comments
 (0)