Open
Description
Describe the bug
transformers_version sometimes requires format X.Y.Z, sometimes X.Y
To reproduce
This works
huggingface_model = HuggingFaceModel(
model_data=s3_location,
role=role,
transformers_version="4.26",
pytorch_version="1.13",
py_version='py39',
)
This doesn't
huggingface_model = HuggingFaceModel(
model_data=s3_location,
role=role,
transformers_version="4.37", # this must be 4.37.0 instead
pytorch_version="2.1.0",
py_version='py310',
)
Expected behavior
Consistent behavior. Would be great to point directly to the website which contains all available images as part of the error message as well. https://github.com/aws/deep-learning-containers/blob/master/available_images.md#huggingface-inference-containers
Screenshots or logs
If applicable, add screenshots or logs to help explain your problem.
System information
A description of your system. Please provide:
- SageMaker Python SDK version:
- Framework name (eg. PyTorch) or algorithm (eg. KMeans): pytorch
- Framework version: varies
- Python version: varies
- CPU or GPU: N/A
- Custom Docker image (Y/N): N
Additional context
None