transformers
6daa9cb5 - add GPTNeoXForSequenceClassification (#22671)

Commit

2 years ago

add GPTNeoXForSequenceClassification (#22671) * add GPTNeoXForSequenceClassification * move the labels to logits.device (ref: #22561) * fix

References

Author

Asugawara

Parents