# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("autonomous019/distilbert_ell3")
model = AutoModelForSequenceClassification.from_pretrained("autonomous019/distilbert_ell3", device_map="auto")Quick Links
a lightweight solution for the Kaggle ELL competition using distilbert
Info about the Kaggle ELL competition: https://www.kaggle.com/competitions/feedback-prize-english-language-learning/code
- Downloads last month
- 10
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="autonomous019/distilbert_ell3")