diff --git a/src/axolotl/evaluate.py b/src/axolotl/evaluate.py index 6d6813730..2b5869939 100644 --- a/src/axolotl/evaluate.py +++ b/src/axolotl/evaluate.py @@ -7,7 +7,6 @@ from pathlib import Path from typing import Dict, Optional import torch -from accelerate.logging import get_logger from datasets import Dataset from transformers.trainer import Trainer @@ -17,6 +16,7 @@ from axolotl.train import ( ) from axolotl.utils.dict import DictDefault from axolotl.utils.distributed import cleanup_distributed +from axolotl.utils.logging import get_logger from axolotl.utils.trainer import setup_trainer project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))