master
test_vision_transformer_creation
def test_vision_transformer_creation(self, mlp_dim, num_heads, num_layers,