How to use dandelin/vilt-b32-mlm with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="dandelin/vilt-b32-mlm")
# Load model directly from transformers import AutoProcessor, AutoModelForMaskedLM processor = AutoProcessor.from_pretrained("dandelin/vilt-b32-mlm") model = AutoModelForMaskedLM.from_pretrained("dandelin/vilt-b32-mlm")
fix "device" and "pixel_values" not defined
· Sign up or log in to comment
