How to deploy Sagemaker Multi-model Endpoints on GPU?


I’m trying to deploy Multi-model Endpoints by following : AWS-SageMaker-Examples/03_MultiModelEndpointWithHuggingFace/huggingface-sagemaker-multi-model-endpoint.ipynb at main · vinayak-shanawad/AWS-SageMaker-Examples · GitHub

I can deploy each model separately either on CPU or GPU.
I can deploy both model on a multi-model Endpoints on CPU, but as soon as I try to deploy them on GPU I get the following error :

ClientError: An error occurred (ValidationException) when calling the CreateModel operation: Your Ecr Image does not contain required com.amazonaws.sagemaker.capabilities.multi-models=true Docker label(s)

Any help would be appreciated !