why would you want to add init to the output of the attention block ?
why would you want to add init to the output of the attention block ?