@@ -321,21 +321,23 @@ def _create_sagemaker_model(
321
321
"""Create a SageMaker Model Entity
322
322
323
323
Args:
324
- instance_type (str): The EC2 instance type that this Model will be
324
+ instance_type (str): Optional. The EC2 instance type that this Model will be
325
325
used for, this is only used to determine if the image needs GPU
326
- support or not.
327
- accelerator_type (str): Type of Elastic Inference accelerator to
326
+ support or not. (Default: None).
327
+ accelerator_type (str): Optional. Type of Elastic Inference accelerator to
328
328
attach to an endpoint for model loading and inference, for
329
329
example, 'ml.eia1.medium'. If not specified, no Elastic
330
- Inference accelerator will be attached to the endpoint.
330
+ Inference accelerator will be attached to the endpoint. (Default: None).
331
331
tags (List[dict[str, str]]): Optional. The list of tags to add to
332
332
the model. Example: >>> tags = [{'Key': 'tagname', 'Value':
333
333
'tagvalue'}] For more information about tags, see
334
334
https://boto3.amazonaws.com/v1/documentation
335
335
/api/latest/reference/services/sagemaker.html#SageMaker.Client.add_tags
336
+ (Default: None).
336
337
serverless_inference_config (sagemaker.serverless.ServerlessInferenceConfig):
337
- Specifies configuration related to serverless endpoint. Instance type is
338
+ Optional. Specifies configuration related to serverless endpoint. Instance type is
338
339
not provided in serverless inference. So this is used to find image URIs.
340
+ (Default: None).
339
341
kwargs: Keyword arguments coming from the caller. This class does not require
340
342
any so they are ignored.
341
343
"""
0 commit comments