diff --git a/src/art/serverless/backend.py b/src/art/serverless/backend.py index 4da5fad5..abf67f69 100644 --- a/src/art/serverless/backend.py +++ b/src/art/serverless/backend.py @@ -481,6 +481,10 @@ async def _train_sft( elif event.type == "training_ended": if pbar is not None: pbar.close() + # Record provenance on the latest W&B artifact for SFT training. + wandb_run = model._get_wandb_run() + if wandb_run is not None: + record_provenance(wandb_run, "serverless-sft") return elif event.type == "training_failed": if pbar is not None: