Helsinki-NLP/opus-100
Viewer • Updated • 55.1M • 24.7k • 259
How to use ehyxu/nmt-corrector-ur2en with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("ehyxu/nmt-corrector-ur2en")
model = AutoModelForSeq2SeqLM.from_pretrained("ehyxu/nmt-corrector-ur2en", device_map="auto")The checkpoint is from Urdu translate to English model of the paper. More information pleace refer this repository.
pre-trained model: mbart-large-50
train dataset: opus-100