diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/models.py b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
index 65e940e58c..05021ed9d2 100644
--- a/sagemaker-core/src/sagemaker/core/jumpstart/models.py
+++ b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
@@ -395,6 +395,9 @@ class CapabilityEnum(StrEnum):
     HYPERPOD_DEPLOYMENT = "HYPERPOD_DEPLOYMENT"
     SAGEMAKER_EVALUATE = "SAGEMAKER_EVALUATE"
     SAGEMAKER_MODEL_OPTIMIZE = "SAGEMAKER_MODEL_OPTIMIZE"
+    TRAINING = "TRAINING"
+    FINE_TUNING = "FINE_TUNING"
+    VALIDATION = "VALIDATION"
 
 
 class DemoNotebookModel(BaseConfig):