diff --git a/src/axolotl/utils/data/sft.py b/src/axolotl/utils/data/sft.py index 70543d7c8..adfd1bd02 100644 --- a/src/axolotl/utils/data/sft.py +++ b/src/axolotl/utils/data/sft.py @@ -1,10 +1,10 @@ """data handling specific to SFT""" import functools +import logging from pathlib import Path from typing import List, Optional, Tuple, Union -from accelerate.logging import get_logger from datasets import ( Dataset, DatasetDict, @@ -55,7 +55,7 @@ process_datasets_for_packing, ) -LOG = get_logger("axolotl") +LOG = logging.getLogger("axolotl.utils.data.sft") @retry_on_request_exceptions(max_retries=3, delay=5)