From 786e54405bf6f9f9a4916d8ea9b7a30dd69b5839 Mon Sep 17 00:00:00 2001 From: Reginald-King Date: Fri, 26 Jun 2026 17:25:36 +0000 Subject: [PATCH] Add explicit model-customization-client-origin tag to all trainer templates Add tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}] to all fine-tuning code templates (SFT, DPO, RLVR, RLAIF builtin, RLAIF custom prompt) so training jobs created via this plugin are tagged with the sherpa-plugin source identifier. --- plugins/sagemaker-ai/skills/finetuning/code_templates/dpo.py | 3 ++- .../skills/finetuning/code_templates/rlaif_builtin.py | 1 + .../skills/finetuning/code_templates/rlaif_custom_prompt.py | 1 + plugins/sagemaker-ai/skills/finetuning/code_templates/rlvr.py | 3 ++- plugins/sagemaker-ai/skills/finetuning/code_templates/sft.py | 3 ++- 5 files changed, 8 insertions(+), 3 deletions(-) diff --git a/plugins/sagemaker-ai/skills/finetuning/code_templates/dpo.py b/plugins/sagemaker-ai/skills/finetuning/code_templates/dpo.py index 6402dcc4..c7e967c7 100644 --- a/plugins/sagemaker-ai/skills/finetuning/code_templates/dpo.py +++ b/plugins/sagemaker-ai/skills/finetuning/code_templates/dpo.py @@ -74,7 +74,8 @@ s3_output_path=S3_OUTPUT_PATH, sagemaker_session=sagemaker_session, #accept_eula=ACCEPT_EULA, # Uncomment for Meta models - role=ROLE_ARN + role=ROLE_ARN, + tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}] ) print("Here are the recommended hyperparameters for the current training job:") print(f"Batch size: {trainer.hyperparameters.global_batch_size}") diff --git a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_builtin.py b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_builtin.py index f6c12e96..cea6e565 100644 --- a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_builtin.py +++ b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_builtin.py @@ -82,6 +82,7 @@ sagemaker_session=sagemaker_session, #accept_eula=ACCEPT_EULA, # Uncomment for Meta models role=ROLE_ARN, + tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}], ) print("Here are the recommended hyperparameters for the current training job:") diff --git a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_custom_prompt.py b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_custom_prompt.py index e603c889..9338bcd7 100644 --- a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_custom_prompt.py +++ b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlaif_custom_prompt.py @@ -95,6 +95,7 @@ sagemaker_session=sagemaker_session, #accept_eula=ACCEPT_EULA, # Uncomment for Meta models role=ROLE_ARN, + tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}], ) print("Here are the recommended hyperparameters for the current training job:") diff --git a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlvr.py b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlvr.py index 1652dad1..07adcdcb 100644 --- a/plugins/sagemaker-ai/skills/finetuning/code_templates/rlvr.py +++ b/plugins/sagemaker-ai/skills/finetuning/code_templates/rlvr.py @@ -89,7 +89,8 @@ sagemaker_session=sagemaker_session, #accept_eula=ACCEPT_EULA, # Uncomment for Meta models role=ROLE_ARN, - custom_reward_function=CUSTOM_REWARD_FUNCTION + custom_reward_function=CUSTOM_REWARD_FUNCTION, + tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}] ) print("Here are the recommended hyperparameters for the current training job:") print(f"Batch size: {trainer.hyperparameters.global_batch_size}") diff --git a/plugins/sagemaker-ai/skills/finetuning/code_templates/sft.py b/plugins/sagemaker-ai/skills/finetuning/code_templates/sft.py index 3b7b1e7c..a867e53f 100644 --- a/plugins/sagemaker-ai/skills/finetuning/code_templates/sft.py +++ b/plugins/sagemaker-ai/skills/finetuning/code_templates/sft.py @@ -75,7 +75,8 @@ s3_output_path=S3_OUTPUT_PATH, sagemaker_session=sagemaker_session, #accept_eula=ACCEPT_EULA, # Uncomment for Meta models - role=ROLE_ARN + role=ROLE_ARN, + tags=[{"Key": "aws-sagemaker:model-customization-client-origin", "Value": "sherpa-plugin"}] ) print("Here are the recommended hyperparameters for the current training job:")