support update only first attention head
