riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-27 00:09:00 +06:00
Code Issues Actions 27 Packages Projects Releases Wiki Activity
51ab25e293
transformers/tests/models/gpt_neox
History
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