From 7e31aa7bd775eaf1817125669442b19f4dde0a33 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 31 Oct 2025 14:43:31 +0700 Subject: [PATCH] fix: duplicate plugin_manager calls --- src/axolotl/train.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/axolotl/train.py b/src/axolotl/train.py index 6cbcb9aec..cce3b8a6a 100644 --- a/src/axolotl/train.py +++ b/src/axolotl/train.py @@ -212,8 +212,7 @@ def execute_training( LOG.info("Starting trainer...") 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( @@ -539,9 +538,6 @@ def setup_model_and_trainer( ) PLUGIN_MANAGER.post_trainer_create(cfg, trainer) - plugin_manager = PluginManager.get_instance() - plugin_manager.post_trainer_create(cfg, trainer) - if cfg.use_ray: try: import ray.train.huggingface.transformers