generative-newsai/news-unmasked
Viewer • Updated • 61.2k • 226 • 1
How to use Qilex/roBERTaNYTheadlines with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="Qilex/roBERTaNYTheadlines") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("Qilex/roBERTaNYTheadlines")
model = AutoModelForMaskedLM.from_pretrained("Qilex/roBERTaNYTheadlines")This is the RoBERTa-large checkpoint domain-adapted on rougly 41000 recent New York Times headlines to fill masks like NYT might.