jhu-clsp/jfleg
Viewer • Updated • 1.5k • 3.87k • 77
How to use JohnGorri/gramercy-ml-gec-small with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("JohnGorri/gramercy-ml-gec-small")
model = AutoModelForSeq2SeqLM.from_pretrained("JohnGorri/gramercy-ml-gec-small", device_map="auto")Base model
google/flan-t5-small