tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

use state for wandb in callbacks (#2930) [skip ci]

Browse Source
This commit is contained in:
Wing Lian
2025-07-17 15:31:56 -04:00
committed by GitHub
parent 8e41317250
commit d23f972602
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/callbacks/__init__.py
View File

@@ -798,7 +798,7 @@ class SaveAxolotlConfigtoWandBCallback(TrainerCallback):
control: TrainerControl, control: TrainerControl,
**kwargs, # pylint: disable=unused-argument **kwargs, # pylint: disable=unused-argument
): ):
if is_main_process(): if state.is_world_process_zero:
try: 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. # 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( with NamedTemporaryFile(