Skip to content

Inconsistent Versioning of transformers_version #4710

Open
@wistuba

Description

@wistuba

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

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions