Before revision:  One key factor to the success of these models is the multi-head attention mechanism.
After revision:  One key factor in the success of these models is the multi-head attention mechanism.