From 1536ee343db2bbebcf2cb191982e112128844406 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 9 Apr 2024 00:56:56 -0400 Subject: [PATCH] also use wandb.save for axolotl yaml --- src/axolotl/utils/callbacks/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py index 644f46984f..09bff1cfc6 100644 --- a/src/axolotl/utils/callbacks/__init__.py +++ b/src/axolotl/utils/callbacks/__init__.py @@ -751,6 +751,7 @@ def on_train_begin( at = wandb.Artifact(f"config-{wandb.run.id}", type="axolotl-config") at.add_file(temp_file.name) wandb.log_artifact(at) + wandb.save(temp_file.name) LOG.info( "The Axolotl config has been saved to the WandB run under files." )