from .modeling_vit import gaudi_vit_self_attention_forward