vision_transformer.py 23.6 KB