diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/models.py b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
index 1e87627821..8b1b282224 100644
--- a/sagemaker-core/src/sagemaker/core/jumpstart/models.py
+++ b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
@@ -129,6 +129,7 @@ class InferenceAmiVersionEnum(StrEnum):
     """InferenceAmiVersionEnum"""
 
     AL2_AMI_SAGEMAKER_INFERENCE_GPU_2 = "al2-ami-sagemaker-inference-gpu-2"
+    AL2_AMI_SAGEMAKER_INFERENCE_GPU_3_1 = "al2-ami-sagemaker-inference-gpu-3-1"
 
 
 class ScopeEnum(StrEnum):
diff --git a/sagemaker-core/src/sagemaker/core/workflow/utilities.py b/sagemaker-core/src/sagemaker/core/workflow/utilities.py
index 18ea133100..007520778e 100644
--- a/sagemaker-core/src/sagemaker/core/workflow/utilities.py
+++ b/sagemaker-core/src/sagemaker/core/workflow/utilities.py
@@ -172,9 +172,9 @@ def get_code_hash(step: Entity) -> str:
         source_code = model_trainer.source_code
         if source_code:
             source_dir = source_code.source_dir
-            dependencies = source_code.dependencies
+            requirements = source_code.requirements
             entry_point = source_code.entry_script
-            return get_training_code_hash(entry_point, source_dir, dependencies)
+            return get_training_code_hash(entry_point, source_dir, requirements)
     return None