Downloads · 30 days
33
1% of all-time downloads
nmt-doc-en-is-2022-10
mideind/nmt-doc-en-is-2022-10
nmt-doc-en-is-2022-10 is a translation model from mideind. Use it when you need text moved from one language to another. It is set up for transformers.
This model was trained to translate multiple sentences at once, compared to one sentence at a time.
All-time downloads
2.4K
Public
Parameters
611M
4.9 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2.4 GB · 50%
At a glance
- Task
- Translation
- Library
- transformers
- Model type
- mbart
- Access
- Public
- Created
- Sep 30, 2022
- Updated
- Apr 2, 2023
- SHA
- aacbf16f
Try a prompt
Model card
From the Hugging Face model README
mBART based translation model
This model was trained to translate multiple sentences at once, compared to one sentence at a time.
It will occasionally combine sentences or add an extra sentence.
This is the same model as are provided on CLARIN: https://repository.clarin.is/repository/xmlui/handle/20.500.12537/278
You can use the following example to get started:
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
import torch
device = torch.cuda.current_device() if torch.cuda.is_available() else -1
tokenizer = AutoTokenizer.from_pretrained("mideind/nmt-doc-en-is-2022-10",src_lang="en_XX",tgt_lang="is_IS")
model = AutoModelForSeq2SeqLM.from_pretrained("mideind/nmt-doc-en-is-2022-10")
translate = pipeline("translation_XX_to_YY",model=model,tokenizer=tokenizer,device=device,src_lang="en_XX",tgt_lang="is_IS")
target_seq = translate("I am using a translation model to translate text from English to Icelandic.",src_lang="en_XX",tgt_lang="is_IS",max_length=128)
print(target_seq[0]['translation_text'].strip('YY '))
- Task
- Translation
- Library
- transformers
- Type
- mbart
- Languages
- en, is, multilingual
- Created
- Sep 30, 2022
- Updated
- Apr 2, 2023