Get a key error if using AutoModelForImageClassification.from_pretrained as the config file/model definition of google/vit-hybrid-base-bit-384 defines model_type as vit-hybrid but the class mapping in transformers.models.auto.configuration_auto uses the key vit_hybrid.
Can make an issue/PR about this to transformers at some point but just leaving this here as a reminder for now.
Get a key error if using
AutoModelForImageClassification.from_pretrainedas the config file/model definition ofgoogle/vit-hybrid-base-bit-384definesmodel_typeasvit-hybridbut the class mapping intransformers.models.auto.configuration_autouses the keyvit_hybrid.Can make an issue/PR about this to transformers at some point but just leaving this here as a reminder for now.