vision_transformer.py 24.4 KB