Browse Source
add en-ja model.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
2 years ago
1 changed files with
12 additions and
0 deletions
-
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 |
|
|
|
|
