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")