transformers/tests/models/gpt_neox
Joao Gante 441de62f49
RoPE models: add numerical sanity-check test for RoPE scaling (#29808)
* add hard rope scaling test

* make fixup

* quick rope scaling tests

* add copy statements
2024-03-28 11:25:50 +00:00
..
__init__.py [WIP] Adding GPT-NeoX-20B (#16659) 2022-05-24 09:31:10 -04:00
test_modeling_gpt_neox.py RoPE models: add numerical sanity-check test for RoPE scaling (#29808) 2024-03-28 11:25:50 +00:00