diff --git a/src/axolotl/integrations/llm_compressor/__init__.py b/src/axolotl/integrations/llm_compressor/__init__.py index 4ed55ead6..d4797b7c2 100644 --- a/src/axolotl/integrations/llm_compressor/__init__.py +++ b/src/axolotl/integrations/llm_compressor/__init__.py @@ -14,7 +14,7 @@ from transformers.trainer import Trainer from transformers.trainer_callback import TrainerCallback, TrainerControl, TrainerState from transformers.training_args import TrainingArguments -from ..base import BasePlugin +from axolotl.integrations.base import BasePlugin P = ParamSpec("P") # Params for generic function signatures R = TypeVar("R") # Return type for generic function signatures