From 76bb09784dbad6f376fc27fa82107a054eef9c06 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 5 Mar 2025 14:05:27 -0500 Subject: [PATCH] fix import --- src/axolotl/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/train.py b/src/axolotl/train.py index aa10d0e06..178b90f7b 100644 --- a/src/axolotl/train.py +++ b/src/axolotl/train.py @@ -15,7 +15,7 @@ from accelerate.logging import get_logger from accelerate.utils import save_fsdp_model from datasets import Dataset from peft import PeftConfig, PeftModel -from transformers import PreTrainedModel, PreTrainedTokenizer, ProcessorMixin, Trainer +from transformers import PreTrainedModel, PreTrainedTokenizer, ProcessorMixin from transformers.integrations.deepspeed import is_deepspeed_zero3_enabled from transformers.trainer import Trainer