Nllb 200 Distilled 600M En Zh CN

🚀 モデルドキュメント：NLLB - 200 - distilled - 600Mによる英語から簡体字中国語への翻訳

このモデルは、MetaのNLLB - 200 - distilled - 600Mをファインチューニングした、英語から簡体字中国語への機械翻訳モデルです。HackerMonica/nllb - 200 - distilled - 600M - en - zh_CNにホストされており、NLLB - 200モデルの蒸留バージョンを使用し、英語と簡体字中国語間の翻訳タスクに特化して最適化されています。

🚀 クイックスタート

このモデルを使用するには、まず必要な依存関係をインストールし、モデルとトークナイザーを初期化します。その後、提供された関数を使用して英語から簡体字中国語への翻訳を行うことができます。

✨ 主な機能

• MetaのNLLB - 200 - distilled - 600Mをベースにファインチューニングされた英語から簡体字中国語への翻訳モデル。
• Hugging Faceのtransformersライブラリを利用して簡単に使用できる。

📦 インストール

モデルを使用するには、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')

高度な使用法

以下の関数を使用して、英語のテキストを簡体字中国語に翻訳することができます。

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

📄 ライセンス

このモデルはCC - BY - NC - 4.0ライセンスの下で提供されています。