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	[WIP] Adding GPT-NeoX-20B (#16659)	2022-05-24 09:31:10 -04:00
test_modeling_gpt_neox.py	Fix mask creations of GPTNeoX and GPT2 (#31944)	2024-07-23 10:11:12 +02:00