# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("moshew/mpnet-base-sst2-distilled")
model = AutoModelForSequenceClassification.from_pretrained("moshew/mpnet-base-sst2-distilled", device_map="auto")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
{'test_accuracy': 0.9426605504587156, 'test_loss': 0.1693699210882187, 'test_runtime': 1.7713, 'test_samples_per_second': 492.29, 'test_steps_per_second': 3.952}
- Downloads last month
- 7
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="moshew/mpnet-base-sst2-distilled")