RoBERTA Action+Context CLS v1 3be98e3
Sebastin Santy commited on
How to use moralstories/roberta-large_action-context with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="moralstories/roberta-large_action-context") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("moralstories/roberta-large_action-context")
model = AutoModelForSequenceClassification.from_pretrained("moralstories/roberta-large_action-context", device_map="auto")