Set more precise shape to the attention weights and outputs b2c5167 ivanzhouyq commited on May 29, 2023