@@ -85,7 +85,8 @@ def _validate_pt_tf_versions(pytorch_version, tensorflow_version, image_uri):
85
85
86
86
87
87
class HuggingFaceModel (FrameworkModel ):
88
- """A Hugging Face SageMaker ``Model`` that can be deployed to an Amazon SageMaker ``Endpoint``."""
88
+ """A Hugging Face SageMaker ``Model`` that can be deployed
89
+ to an Amazon SageMaker ``Endpoint``."""
89
90
90
91
_framework_name = "huggingface"
91
92
@@ -197,14 +198,16 @@ def register(
197
198
transform_instances (list): A list of the instance types on which a transformation
198
199
job can be run or on which an endpoint can be deployed.
199
200
model_package_name (str): Model Package name, exclusive to `model_package_group_name`,
200
- using `model_package_name` makes the Model Package un-versioned. Defaults to ``None``.
201
+ using `model_package_name` makes the Model Package un-versioned.
202
+ Defaults to ``None``.
201
203
model_package_group_name (str): Model Package Group name, exclusive to
202
204
`model_package_name`, using `model_package_group_name` makes the Model Package
203
205
versioned. Defaults to ``None``.
204
206
image_uri (str): Inference image URI for the container. Model class' self.image will
205
207
be used if it is None. Defaults to ``None``.
206
208
model_metrics (ModelMetrics): ModelMetrics object. Defaults to ``None``.
207
- metadata_properties (MetadataProperties): MetadataProperties object. Defaults to ``None``.
209
+ metadata_properties (MetadataProperties): MetadataProperties object.
210
+ Defaults to ``None``.
208
211
marketplace_cert (bool): A boolean value indicating if the Model Package is certified
209
212
for AWS Marketplace. Defaults to ``False``.
210
213
approval_status (str): Model Approval Status, values can be "Approved", "Rejected",
0 commit comments