许可协议: cc-by-nc-4.0
支持语言:

• 英文
• 中文
  评估指标:
• bleu
  任务类型: 翻译

模型文档：基于NLLB-200-distilled-600M的英译简中模型

模型概述

本文档描述了一个基于Meta公司NLLB-200-distilled-600M模型微调的机器翻译模型，专门用于英语到简体中文的翻译任务。该模型托管在HackerMonica/nllb-200-distilled-600M-en-zh_CN，采用了NLLB-200模型的蒸馏版本，并针对英汉翻译任务进行了专项优化。

依赖环境

模型需要安装Hugging Face的transformers库：

pip install transformers  

环境配置

从transformers库导入必要组件：

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer  

初始化模型与分词器：

model = AutoModelForSeq2SeqLM.from_pretrained('HackerMonica/nllb-200-distilled-600M-en-zh_CN')  
tokenizer = AutoTokenizer.from_pretrained('HackerMonica/nllb-200-distilled-600M-en-zh_CN')  

使用方式

通过以下Python代码实现英译中文本转换：

def translate(text):  
    inputs = tokenizer(text, return_tensors="pt").to("cuda")  
    translated_tokens = model.generate(  
        **inputs, forced_bos_token_id=tokenizer.lang_code_to_id["zho_Hans"], max_length=300  
    )  
    translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]  
    return translation