fast follow on js uncompressed support

Author: EC2 Default User

src/sagemaker/serve/model_server/tgi/prepare.py

@@ -17,16 +17,16 @@ def _copy_jumpstart_artifacts(model_data: str, js_id: str, code_dir: Path) -> bo
     """Placeholder Docstring"""
     logger.info("Downloading JumpStart artifacts from S3...")
     with _tmpdir(directory=str(code_dir)) as js_model_dir:
-        # TODO: remove if block after 10/30 once everything is shifted to uncompressed
-        if model_data.endswith("tar.gz"):
-            subprocess.run(["aws", "s3", "cp", model_data, js_model_dir])
-
+        js_model_data_loc = model_data.get("S3DataSource").get("S3Uri")
+        # TODO: leave this check here until we are sure every js model has moved to uncompressed
+        if js_model_data_loc.endswith("tar.gz"):
+            subprocess.run(["aws", "s3", "cp", js_model_data_loc, js_model_dir])
             logger.info("Uncompressing JumpStart artifacts for faster loading...")
             tmp_sourcedir = Path(js_model_dir).joinpath(f"infer-prepack-{js_id}.tar.gz")
             with tarfile.open(str(tmp_sourcedir)) as resources:
                 resources.extractall(path=code_dir)
         else:
-            subprocess.run(["aws", "s3", "cp", model_data, js_model_dir, "--recursive"])
+            subprocess.run(["aws", "s3", "cp", js_model_data_loc, js_model_dir, "--recursive"])
     return True