tests/models/gpt2/test_modeling_gpt2.py · 605f3245dcca34381c35520c35ba0b701ed80d58 · chenpangpang / transformers

Fix mask creations of `GPTNeoX` and `GPT2` (#31944) · 605f3245

Anton Vlasjuk authored Jul 23, 2024

* fix mask creation of gpt2 and gpt_neox caused by me

* forgot the reshape of masks when shape > 2

* add tests for gpt neox and gpt2

* nit on a comment

605f3245

test_modeling_gpt2.py 40.5 KB

Replace test_modeling_gpt2.py