test_model_loading.py

from transformers import AutoModel, AutoConfig

# Load Config and Model
config = AutoConfig.from_pretrained("shravvvv/SAG-ViT")
model = AutoModel.from_pretrained("shravvvv/SAG-ViT", config=config)

# Check Model
print(model)