From d23f972602ac50882c123a91210547e9c9f3bfd8 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 17 Jul 2025 15:31:56 -0400 Subject: [PATCH] use state for wandb in callbacks (#2930) [skip ci] --- src/axolotl/utils/callbacks/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py index bb777fc90..b0738bf27 100644 --- a/src/axolotl/utils/callbacks/__init__.py +++ b/src/axolotl/utils/callbacks/__init__.py @@ -798,7 +798,7 @@ class SaveAxolotlConfigtoWandBCallback(TrainerCallback): control: TrainerControl, **kwargs, # pylint: disable=unused-argument ): - if is_main_process(): + if state.is_world_process_zero: try: # sync config to top level in run, cannot delete file right away because wandb schedules it to be synced even w/policy = 'now', so let OS delete it later. with NamedTemporaryFile(