Anton Vlasjuk 605f3245dc Fix mask creations of GPTNeoX and GPT2 (#31944) ...

* 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

2024-07-23 10:11:12 +02:00

..

__init__.py

…

test_modeling_gpt_neox.py

Fix mask creations of GPTNeoX and GPT2 (#31944)

2024-07-23 10:11:12 +02:00