transformer_s2_layer.py 26.8 KB