vision_transformer.py 15.0 KB