🚀 模型文档:使用NLLB - 200进行沃洛夫语到法语的翻译
本模型是一个经过微调的机器翻译模型,基于Meta的NLLB - 200,专门用于沃洛夫语和法语之间的翻译。它能够有效提升这两种语言之间的翻译质量,为跨语言交流提供有力支持。
🚀 快速开始
依赖安装
本模型依赖Hugging Face的transformers
库,请确保你已安装该库:
pip install transformers
环境设置
从transformers
库导入必要的类:
from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
初始化模型和分词器:
model = AutoModelForSeq2SeqLM.from_pretrained('cifope/nllb-200-wo-fr-distilled-600M')
tokenizer = NllbTokenizer.from_pretrained('facebook/nllb-200-distilled-600M')
✨ 主要特性
- 基于Meta的NLLB - 200模型进行微调,针对沃洛夫语和法语的翻译任务进行了优化。
- 提供了两种翻译方向的函数,支持沃洛夫语到法语以及法语到沃洛夫语的翻译。
💻 使用示例
基础用法
法语到沃洛夫语的翻译
def translate(text, src_lang='fra_Latn', tgt_lang='wol_Latn', a=16, b=1.5, max_input_length=1024, **kwargs):
tokenizer.src_lang = src_lang
tokenizer.tgt_lang = tgt_lang
inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=max_input_length)
result = model.generate(
**inputs.to(model.device),
forced_bos_token_id=tokenizer.convert_tokens_to_ids(tgt_lang),
max_new_tokens=int(a + b * inputs.input_ids.shape[1]),
**kwargs
)
return tokenizer.batch_decode(result, skip_special_tokens=True)
french_text = "L'argent peut être échangé à la seule banque des îles située à Stanley"
wolof_translation = translate(french_text)
print(wolof_translation)
沃洛夫语到法语的翻译
def reversed_translate(text, src_lang='wol_Latn', tgt_lang='fra_Latn', a=16, b=1.5, max_input_length=1024, **kwargs):
tokenizer.src_lang = src_lang
tokenizer.tgt_lang = tgt_lang
inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=max_input_length)
result = model.generate(
**inputs.to(model.device),
forced_bos_token_id=tokenizer.convert_tokens_to_ids(tgt_lang),
max_new_tokens=int(a + b * inputs.input_ids.shape[1]),
**kwargs
)
return tokenizer.batch_decode(result, skip_special_tokens=True)
wolof_text = "alkaati yi tàmbali nañu xàll léegi kilifa gi ñów"
french_translation = reversed_translate(wolof_text)
print(french_translation)
高级用法
沃洛夫语到英语的翻译
wolof_text = "alkaati yi tàmbali nañu xàll léegi kilifa gi ñów"
english_translation = reversed_translate(wolof_text,tgt_lang="eng_Latn")
print(english_translation)
📄 许可证
本模型采用MIT许可证。