Transformers How to use ydshieh/tiny-random-ViTForMaskedImageModeling with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, ViTForMaskedImageModeling
processor = AutoImageProcessor.from_pretrained("ydshieh/tiny-random-ViTForMaskedImageModeling")
model = ViTForMaskedImageModeling.from_pretrained("ydshieh/tiny-random-ViTForMaskedImageModeling", device_map="auto") 
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, ViTForMaskedImageModeling processor = AutoImageProcessor.from_pretrained("ydshieh/tiny-random-ViTForMaskedImageModeling") model = ViTForMaskedImageModeling.from_pretrained("ydshieh/tiny-random-ViTForMaskedImageModeling", device_map="auto")