diff --git a/src/sagemaker/image_uri_config/neo-tensorflow.json b/src/sagemaker/image_uri_config/neo-tensorflow.json
index 8fe391b43d..3c76bd68ac 100644
--- a/src/sagemaker/image_uri_config/neo-tensorflow.json
+++ b/src/sagemaker/image_uri_config/neo-tensorflow.json
@@ -13,6 +13,9 @@
         "1.12.0": "1.15.3",
         "1.13.0": "1.15.3",
         "1.14.0": "1.15.3",
+        "2.4": "2.4.2",
+        "2.4.0": "2.4.2",
+        "2.4.1": "2.4.2",
         "2.4.2": "2.4.2"
     },
     "versions": {
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
index b8c6b4ea22..317e865ddd 100644
--- a/src/sagemaker/model.py
+++ b/src/sagemaker/model.py
@@ -630,6 +630,8 @@ def _compilation_job_config(
 
         if (
             framework.lower() == "pytorch"
+            or framework.lower() == "tensorflow"
+            and target_instance_type is not None
             and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None
             and framework_version is not None
         ):