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

docs/source/en
- model_doc
  - gpt_neox.mdx
- tasks
  - sequence_classification.mdx
src/transformers
- __init__.py
- models
  - auto
    - modeling_auto.py
  - gpt_neox
    - __init__.py
    - modeling_gpt_neox.py
- utils
  - dummy_pt_objects.py
tests/models/gpt_neox
- test_modeling_gpt_neox.py