riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-24 14:58:56 +06:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity
9435cc6670
transformers/tests/models/gpt_neox
History
Sugawara 6daa9cb515
add GPTNeoXForSequenceClassification (#22671) 
* add GPTNeoXForSequenceClassification

* move the labels to logits.device (ref: #22561)

* fix
2023-04-10 11:52:23 -04:00
..
__init__.py [WIP] Adding GPT-NeoX-20B (#16659) 2022-05-24 09:31:10 -04:00
test_modeling_gpt_neox.py add GPTNeoXForSequenceClassification (#22671) 2023-04-10 11:52:23 -04:00