fix dist state init before deepspeed setup (#2737)

2025-05-28 14:59:57 -04:00
parent 2962a398b7
commit bde8b5b6bd
1 changed files with 1 additions and 1 deletions
--- 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