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:")