jhu-clsp/jfleg
Viewer • Updated • 1.5k • 2.69k • 77
How to use Ragnov/T5-Base-Grammar-Checker with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("Ragnov/T5-Base-Grammar-Checker")
model = AutoModelForSeq2SeqLM.from_pretrained("Ragnov/T5-Base-Grammar-Checker", device_map="auto")