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

add close to comment block

Browse Source
This commit is contained in:
Wing Lian
2025-05-25 16:53:20 -04:00
parent b4e96ef12c
commit 505009b454
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/core/builders/causal.py
View File

@@ -130,6 +130,9 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
return callbacks return callbacks
def _get_trainer_cls(self): def _get_trainer_cls(self):
"""
Gets the trainer class for the given configuration.
"""
if self.cfg.plugins: if self.cfg.plugins:
plugin_manager = PluginManager.get_instance() plugin_manager = PluginManager.get_instance()
trainer_cls = plugin_manager.get_trainer_cls(self.cfg) trainer_cls = plugin_manager.get_trainer_cls(self.cfg)