add GPTNeoXForSequenceClassification (#22671)

* add GPTNeoXForSequenceClassification * move the labels to logits.device (ref: #22561) * fix
2023-04-11 00:52:23 +09:00
parent f74b40208d
commit 6daa9cb515
8 changed files with 175 additions and 6 deletions
--- a/docs/source/en/model_doc/gpt_neox.mdx
+++ b/docs/source/en/model_doc/gpt_neox.mdx
@@ -78,3 +78,8 @@ The `generate()` method can be used to generate text using GPT Neo model.

 [[autodoc]] GPTNeoXForCausalLM
    - forward
+
+## GPTNeoXForSequenceClassification
+
+[[autodoc]] GPTNeoXForSequenceClassification
+    - forward