vision_transformer.py 23.2 KB