version misconfiguration for sagemaker

#12
by marcinp - opened

Hi, i found there is a wrong version of the transformers in the deploy snippet for aws sagemaker.
The snippet:

huggingface_model = HuggingFaceModel(
transformers_version='4.26.0',
pytorch_version='1.13.1',
py_version='py39',
env=hub,
role=role,
)

a config for auto in transformers for a given version (lack of blip-2) :
https://github.com/huggingface/transformers/blob/v4.26.0/src/transformers/models/auto/configuration_auto.py

The config.json contains a proper version of transformers:

https://huggingface.co/Salesforce/blip2-opt-2.7b/blob/main/config.json

Sign up or log in to comment