We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3d8dd4d commit 3c3a589Copy full SHA for 3c3a589
1 file changed
src/art/serverless/backend.py
@@ -481,6 +481,10 @@ async def _train_sft(
481
elif event.type == "training_ended":
482
if pbar is not None:
483
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")
488
return
489
elif event.type == "training_failed":
490
0 commit comments