File size: 225 Bytes
a041fbe
64e8835
a041fbe
 
 
64e8835
a041fbe
64e8835
1
2
3
4
5
6
7
8
9
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)