vision_transformer.py 15.9 KB