also use wandb.save for axolotl yaml

axolotl-ai-cloud · Apr 9, 2024 · 1536ee3 · 1536ee3
1 parent 8414bb0
commit 1536ee3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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."
                 )