How to use theadityamittal/vit_h with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("theadityamittal/vit_h", device_map="auto")
