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

fix: duplicate plugin_manager calls

Browse Source
This commit is contained in:
NanoCode012
2025-10-31 14:43:31 +07:00
parent c1dba2f6df
commit 7e31aa7bd7
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/axolotl/train.py
View File

@@ -212,8 +212,7 @@ def execute_training(
LOG.info("Starting trainer...") LOG.info("Starting trainer...")
trainer.train(resume_from_checkpoint=resume_from_checkpoint) trainer.train(resume_from_checkpoint=resume_from_checkpoint)
plugin_manager = PluginManager.get_instance() PLUGIN_MANAGER.post_train(cfg, trainer.model)
plugin_manager.post_train(cfg, trainer.model)
def save_trained_model( def save_trained_model(
@@ -539,9 +538,6 @@ def setup_model_and_trainer(
) )
PLUGIN_MANAGER.post_trainer_create(cfg, trainer) PLUGIN_MANAGER.post_trainer_create(cfg, trainer)
plugin_manager = PluginManager.get_instance()
plugin_manager.post_trainer_create(cfg, trainer)
if cfg.use_ray: if cfg.use_ray:
try: try:
import ray.train.huggingface.transformers import ray.train.huggingface.transformers