(sagemaker-model-deployment): Support AWS Inferentia instance types

[kukushking]

Describe the feature

I'd like to be able to use AWS Inferentia for my endpoint inference containers

Use Case

Proposed Solution

No response

Other Information

No response

Acknowledgements

• I may be able to implement this feature request
• This feature might incur a breaking change

[krokoko]

Hi @kukushking , we do have 2 samples demonstrating how to deploy models on Inferentia :

• https://github.com/aws-samples/generative-ai-cdk-constructs-samples/tree/main/samples/sagemaker_huggingface_inferentia
• https://github.com/aws-samples/generative-ai-cdk-constructs-samples/tree/main/samples/sagemaker_custom_endpoint

Is there a model in particular you are looking to deploy ?

[kukushking]

Thanks @krokoko. Is this supported for JumpStart Foundation Models? I don't see Inf instance types here

[kukushking]

Ah sorry just found it. Nvm, closing the issue.

[kukushking]

Error: The instance type ml.inf1.2xlarge is not supported. Default instance type: ml.g5.2xlarge. Supported instance types: ml.g5.2xlarge, ml.g5.4xlarge, ml.g5.8xlarge, ml.g5.16xlarge.

I get this error when deploying Mistral 7B with Inf. Am I doing something wrong?

[krokoko]

Could you please share the code snippet you are using ?

[kukushking]

Sure, here is the link.

[krokoko]

Thanks @kukushking , will add it to the backlog