tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix(RL): address plugin rl overwriting trainer_cls (#2697) [skip ci]

Browse Source 
* fix: plugin rl overwrite trainer_cls

* feat(test): add test to catch trainer_cls is not None
This commit is contained in:
NanoCode012
2025-05-22 19:19:12 +07:00
committed by GitHub
parent 1c83a1a020
commit 798b5f5cfd
2 changed files with 28 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/core/trainer_builder.py
View File

@@ -1195,7 +1195,9 @@ class HFRLTrainerBuilder(TrainerBuilderBase):
if self.cfg.plugins:
plugin_manager = PluginManager.get_instance()
trainer_cls = plugin_manager.get_trainer_cls(self.cfg)
temp_trainer_cls = plugin_manager.get_trainer_cls(self.cfg)
if temp_trainer_cls is not None:
trainer_cls = temp_trainer_cls
sig = inspect.signature(trainer_cls)
if "tokenizer" in sig.parameters.keys():