From edda648cd9184073863c37f1fafedbb9f4047f3b Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 9 Apr 2024 11:09:03 -0400 Subject: [PATCH] Update src/axolotl/utils/callbacks/__init__.py --- src/axolotl/utils/callbacks/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py index 09bff1cfc6..edc0f348f8 100644 --- a/src/axolotl/utils/callbacks/__init__.py +++ b/src/axolotl/utils/callbacks/__init__.py @@ -748,9 +748,9 @@ def on_train_begin( mode="w", delete=False, suffix=".yml", prefix="axolotl_config_" ) as temp_file: copyfile(self.axolotl_config_path, temp_file.name) - at = wandb.Artifact(f"config-{wandb.run.id}", type="axolotl-config") - at.add_file(temp_file.name) - wandb.log_artifact(at) + artifact = wandb.Artifact(f"config-{wandb.run.id}", type="axolotl-config") + artifact.add_file(temp_file.name) + wandb.log_artifact(artifact) wandb.save(temp_file.name) LOG.info( "The Axolotl config has been saved to the WandB run under files."