model_vit function

Vision Transformer Implementation