get_model, docstring, and if changes

Author: Bryannah Hernandez

src/sagemaker/serve/builder/transformers_builder.py

@@ -307,7 +307,7 @@ def _get_supported_version(self, hf_config, hugging_face_version, base_fw):
         return sorted(versions_to_return, reverse=True)[0]
 
     def _auto_detect_container(self):
-        """Placeholder docstring"""
+        """Set image_uri by detecting container via model name or inference spec""" 
         # Auto detect the container image uri
         if self.image_uri:
             logger.info(
@@ -350,8 +350,7 @@ def _build_for_transformers(self):
 
         if self.inference_spec:
 
-            if not os.path.exists(self.model_path):
-                os.makedirs(self.model_path)
+            os.makedirs(self.model_path, exist_ok=True)
 
             code_path = Path(self.model_path).joinpath("code")
 

src/sagemaker/serve/spec/inference_spec.py

@@ -30,3 +30,6 @@ def invoke(self, input_object: object, model: object):
 
     def prepare(self, *args, **kwargs):
         """Custom prepare function"""
+
+    def get_model(self):
+     """Return HuggingFace model name for inference spec"""