update

Author: Qingzi-Lan

src/sagemaker/model.py

@@ -620,15 +620,13 @@ def _compilation_job_config(
         }
 
         if (
-            framework.lower() == "pytorch"
+            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
         ):
             input_model_config["FrameworkVersion"] = utils.get_short_version(framework_version)
 
-        if framework.lower() == "tensorflow" and framework_version is not None:
-            input_model_config["FrameworkVersion"] = framework_version
-
         role = self.sagemaker_session.expand_role(role)
         output_model_config = {
             "S3OutputLocation": output_path,