Transformers How to use klhung/ViLT_OCT with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoProcessor, CustomViltForClassification
processor = AutoProcessor.from_pretrained("klhung/ViLT_OCT")
model = CustomViltForClassification.from_pretrained("klhung/ViLT_OCT", device_map="auto") 
