deprecated wandb.save

axolotl-ai-cloud · Apr 5, 2024 · 8414bb0 · 8414bb0
1 parent bf4cd67
commit 8414bb0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py
@@ -748,7 +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)
-                    wandb.save(temp_file.name)
+                    at = wandb.Artifact(f"config-{wandb.run.id}", type="axolotl-config")
+                    at.add_file(temp_file.name)
+                    wandb.log_artifact(at)
                 LOG.info(
                     "The Axolotl config has been saved to the WandB run under files."
                 )