Skip to content

Commit 3c3a589

Browse files
authored
Record serverless SFT provenance on training completion (#590)
* Record serverless SFT provenance on training completion * Remove serverless SFT provenance unit test
1 parent 3d8dd4d commit 3c3a589

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/art/serverless/backend.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,10 @@ async def _train_sft(
481481
elif event.type == "training_ended":
482482
if pbar is not None:
483483
pbar.close()
484+
# Record provenance on the latest W&B artifact for SFT training.
485+
wandb_run = model._get_wandb_run()
486+
if wandb_run is not None:
487+
record_provenance(wandb_run, "serverless-sft")
484488
return
485489
elif event.type == "training_failed":
486490
if pbar is not None:

0 commit comments

Comments
 (0)