feat: call hook_pre_create_trainer for rl

This commit is contained in:
NanoCode012
2025-05-22 17:42:46 +07:00
parent 49888eccb9
commit b87850e11b

View File

@@ -208,6 +208,11 @@ class HFRLTrainerBuilder(TrainerBuilderBase):
trainer_kwargs["dataset_tags"] = [
d["path"] for d in self.cfg.datasets if not Path(d["path"]).is_dir()
]
trainer_kwargs, trainer_cls = self.hook_pre_create_trainer(
trainer_kwargs, trainer_cls
)
trainer = trainer_cls(
*trainer_cls_args,
args=training_args,