refactor: unify the processing of the attention mask #1230
+56
−41
Merged
Loading