Error "ImportError: cannot import name 'ViTModel'"

[RyanHangZhou]

Hi,

With PyTorch 1.0.0 and transformer (from 3.8 to 4.18), for running sample inference, we always have the problem: ImportError: cannot import name 'ViTModel'. The provided environment seems errorous for deployment. It seems that the pretrained model is obtained from a very early PyTorch version (1.0.0), while for installing transformer package, PyTorch 1.0.0 has incompatible issues.

Looking forward to your responses, thanks.