transformers/tests/models/gpt_neox
Arthur 83f9196cc4
[`GPTNeoX`] Fix BC issue with 4.36 (#28602)
* fix dtype issue

* add a test

* update copied from mentions

* nits

* fixup

* fix copies

* Apply suggestions from code review
2024-01-21 17:01:19 +00:00
..
__init__.py [WIP] Adding GPT-NeoX-20B (#16659) 2022-05-24 09:31:10 -04:00
test_modeling_gpt_neox.py [`GPTNeoX`] Fix BC issue with 4.36 (#28602) 2024-01-21 17:01:19 +00:00