make model_data optional

Author: Wang Napieralski

src/sagemaker/huggingface/model.py

@@ -67,6 +67,7 @@ def __init__(
 
 
 def _validate_pt_tf_versions(pytorch_version, tensorflow_version, image_uri):
+    """Placeholder docstring"""
 
     if image_uri is not None:
         return
@@ -84,14 +85,14 @@ def _validate_pt_tf_versions(pytorch_version, tensorflow_version, image_uri):
 
 
 class HuggingFaceModel(FrameworkModel):
-    """An PyTorch SageMaker ``Model`` that can be deployed to a SageMaker ``Endpoint``."""
+    """An HuggingFace SageMaker ``Model`` that can be deployed to a SageMaker ``Endpoint``."""
 
     _framework_name = "huggingface"
 
     def __init__(
         self,
-        model_data,
         role,
+        model_data=None,
         entry_point=None,
         transformers_version=None,
         tensorflow_version=None,
@@ -102,7 +103,7 @@ def __init__(
         model_server_workers=None,
         **kwargs,
     ):
-        """Initialize a PyTorchModel.
+        """Initialize a HuggingFaceModel.
 
         Args:
             model_data (str): The S3 location of a SageMaker model data

src/sagemaker/model.py

@@ -1124,7 +1124,7 @@ def _upload_code(self, key_prefix, repack=False):
                 dependencies=self.dependencies,
             )
 
-        if repack:
+        if repack and self.model_data is not None:
             bucket = self.bucket or self.sagemaker_session.default_bucket()
             repacked_model_data = "s3://" + "/".join([bucket, key_prefix, "model.tar.gz"])