vision_transformer.py 14.6 KB