initial commit 25b2a12
Glenn, Parker commited on
How to use parkervg/destt5-schema-prediction with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("parkervg/destt5-schema-prediction")
model = AutoModelForSeq2SeqLM.from_pretrained("parkervg/destt5-schema-prediction", device_map="auto")