Load ViT with [AutoModelForImageClassification].