corrected to original config models 4f6f57d
Pablo Serna commited on
How to use p-serna/bert2bert-spanish-paraphraser with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("p-serna/bert2bert-spanish-paraphraser")
model = AutoModelForSeq2SeqLM.from_pretrained("p-serna/bert2bert-spanish-paraphraser", device_map="auto")