Transformers How to use Tommidi/st_vit_pretrained-1epoch-ucf101 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import ST_Vit
model = ST_Vit.from_pretrained("Tommidi/st_vit_pretrained-1epoch-ucf101", device_map="auto") 
