Transformers How to use parthpk/mae_small with Transformers:
# Load model directly
from transformers import AutoImageProcessor, AutoModelForPreTraining
processor = AutoImageProcessor.from_pretrained("parthpk/mae_small")
model = AutoModelForPreTraining.from_pretrained("parthpk/mae_small")