vit_model.py 7.92 KB