From bde8b5b6bda0233699aba8ca85949a0b4ac7409d Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Wed, 28 May 2025 14:59:57 -0400 Subject: [PATCH] fix dist state init before deepspeed setup (#2737) --- src/axolotl/integrations/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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