sagemaker not supporting modernBERT trained model with transformers 4.49.0
Hello,
I created modernBERT trained model using latest version of transformers - 4.49.06
and I am trying to deploy this to sagemaker, [I have also upgraded sagemaker to the latest]
create HF中国镜像站 Model Class
huggingface_model = HuggingFaceModel(
model_data=model_data, # configuration for loading model from s3
role=role, # iam role with permissions to create an Endpoint
transformers_version="4.49.06", # transformers version used
pytorch_version="2.4.1", # pytorch version used
py_version='py310', # python version used
env={"HF_TASK": "text-classification"} # Set the NLP task
)
But I get this error:
[ValueError: Unsupported huggingface version: 4.49.06. You may need to upgrade your SDK version](ValueError: Unsupported huggingface version: 4.49.06. You may need to upgrade your SDK version (pip install -U
sagemaker) for newer huggingface versions. Supported huggingface version(s): 4.6.1, 4.10.2, 4.11.0, 4.12.3, 4.17.0,
4.26.0, 4.28.1, 4.37.0, 4.6, 4.10, 4.11, 4.12, 4.17, 4.26, 4.28, 4.37.)
How do I deploy this to sagemaker using HuggingFaceModel.
Thank you for the help.