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