Nllb 200 Wo Fr Distilled 600M

🚀 模型文檔：使用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許可證。