Update src/sagemaker/model.py

Co-authored-by: icywang86rui <[email protected]>

Author: nikhil-sk

src/sagemaker/model.py

@@ -279,7 +279,10 @@ def _compilation_image_uri(self, region, target_instance_type, framework, framew
 
         if framework == "xgboost":
             framework_suffix = "-neo"
+        elif target_instance_typ.startswith("ml_inf"):
+            framework_prefix = "inferentia-"
         else:
+            framework_prefix = "neo-"
             if target_instance_type.startswith("ml_inf"):
                 framework_prefix = "inferentia-"
             else: