From 5184a1f49f46d662b9d498784faa3e3ae9a0e46d Mon Sep 17 00:00:00 2001 From: wxywb Date: Fri, 28 Oct 2022 17:32:43 +0800 Subject: [PATCH] add en-ja model. Signed-off-by: wxywb --- opus_mt.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/opus_mt.py b/opus_mt.py index 9860433..5dd8847 100644 --- a/opus_mt.py +++ b/opus_mt.py @@ -52,4 +52,16 @@ class OpusMT(NNOperator): configs['opus-mt-zh-en'] = {} configs['opus-mt-zh-en']['tokenizer'] = 'Helsinki-NLP/opus-mt-zh-en' configs['opus-mt-zh-en']['model'] = 'Helsinki-NLP/opus-mt-zh-en' + + configs['opus-mt-zh-en'] = {} + configs['opus-mt-zh-en']['tokenizer'] = 'Helsinki-NLP/opus-mt-zh-en' + configs['opus-mt-zh-en']['model'] = 'Helsinki-NLP/opus-mt-zh-en' + + configs['opus-tatoeba-en-ja'] = {} + configs['opus-tatoeba-en-ja']['tokenizer'] = 'Helsinki-NLP/opus-tatoeba-en-ja' + configs['opus-tatoeba-en-ja']['model'] = 'Helsinki-NLP/opus-tatoeba-en-ja' + + configs['opus-tatoeba-ja-en'] = {} + configs['opus-tatoeba-ja-en']['tokenizer'] = 'Helsinki-NLP/opus-tatoeba-ja-en' + configs['opus-tatoeba-ja-en']['model'] = 'Helsinki-NLP/opus-tatoeba-ja-en' return configs