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