vision_transformer.py 23.5 KB