transformers
add GPTNeoXForSequenceClassification
#22671
Merged
Overview
Commits
3
Changes
View On GitHub

add GPTNeoXForSequenceClassification #22671

sgugger merged 3 commits into huggingface:main from Asugawara:add_gptneox_seq_classification
Asugawara
HuggingFaceDocBuilderDev
younesbelkada
younesbelkada approved these changes on 2023-04-10
Asugawara add GPTNeoXForSequenceClassification
2eef5c22
Asugawara Asugawara force pushed to 2eef5c22 2 years ago
sgugger
sgugger approved these changes on 2023-04-10
Asugawara move the labels to logits.device (ref: #22561)
330e7791
Asugawara fix
606726b0
sgugger sgugger merged 6daa9cb5 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum