vit_small.py 4.89 KB