diff --git a/src/axolotl/integrations/base.py b/src/axolotl/integrations/base.py index 11d85f8f8..0edc9fdea 100644 --- a/src/axolotl/integrations/base.py +++ b/src/axolotl/integrations/base.py @@ -32,7 +32,7 @@ from transformers import PreTrainedModel, Trainer from axolotl.utils.dict import DictDefault from axolotl.utils.logging import get_logger -LOG = get_logger(__name__) +LOG = get_logger(__name__, use_environ=True) if TYPE_CHECKING: from axolotl.common.datasets import TrainDatasetMeta