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