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