How to use trl-internal-testing/tiny-GPTNeoXForSequenceClassification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="trl-internal-testing/tiny-GPTNeoXForSequenceClassification")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("trl-internal-testing/tiny-GPTNeoXForSequenceClassification") model = AutoModelForSequenceClassification.from_pretrained("trl-internal-testing/tiny-GPTNeoXForSequenceClassification")
be8b86c cc0b5fe be8b86c cc0b5fe be8b86c cc0b5fe
1
2
3
4
5
6
7
8
9
10
--- library_name: transformers tags: - trl --- # Tiny GPTNeoXForSequenceClassification This is a minimal model built for unit tests in the [TRL](https://github.com/huggingface/trl) library.
