Transformers How to use shehan97/mobilevitv2-2.0-imagenet1k-256 with Transformers:
# Load model directly
from transformers import AutoImageProcessor, MobileViTv2ForImageClassification
processor = AutoImageProcessor.from_pretrained("shehan97/mobilevitv2-2.0-imagenet1k-256")
model = MobileViTv2ForImageClassification.from_pretrained("shehan97/mobilevitv2-2.0-imagenet1k-256", device_map="auto") 
