aws/sagemaker-huggingface-inference-toolkit

Update transformers

segments-tobias opened this issue · 1 comments

Hi! I'd like to deploy a SegFormer model, which was introduced in v4.13.0. However, the Sagemaker toolkit only supports transformers up to v4.12.3.

I already have a working custom inference.py script, but it uses an outdated feature extractor because of the wrong version of transformers, thus leading to incorrect predictions.

Is there a workaround, or when do you expect supporting the newer versions of transformers?

Turns out adding the required transformers version in requirements.txt works!