diff --git a/assets/training/finetune_acft_hf_nlp/components_v2/finetune/chat_completion/spec.yaml b/assets/training/finetune_acft_hf_nlp/components_v2/finetune/chat_completion/spec.yaml
index 1f5d54ee2a..96006e6c23 100644
--- a/assets/training/finetune_acft_hf_nlp/components_v2/finetune/chat_completion/spec.yaml
+++ b/assets/training/finetune_acft_hf_nlp/components_v2/finetune/chat_completion/spec.yaml
@@ -334,6 +334,10 @@ inputs:
     type: string
     optional: True
     description: Name of the registered model
+  model_registration_tag:
+    type: string
+    optional: True
+    description: Tags for registered model
 outputs:
   output_model:
     type: uri_folder