vit_model.py 6.73 KB