How to use Bainbridge/vilt-b32-mlm-mami with Transformers:
# Load model directly from transformers import AutoProcessor, ViltForImageTextClassification processor = AutoProcessor.from_pretrained("Bainbridge/vilt-b32-mlm-mami") model = ViltForImageTextClassification.from_pretrained("Bainbridge/vilt-b32-mlm-mami", device_map="auto")
