Nllb Moe 54b 4bit

🚀 NLLB - MoE

このモデルは、多言語翻訳に特化したモデルで、多くの言語間での翻訳を可能にします。高速なロードと比較的少ないVRAM使用量で動作し、機械翻訳分野での最先端のオープンアクセスモデルとなっています。

🚀 クイックスタート

このモデルを使用するには、bitsandbytesを最新バージョンにアップグレードし、以下のhf transformersのプルリクエストをインストールします。 https://github.com/huggingface/transformers/pull/26037

この方法を使うと、VRAMを37GB使用し、15分かかっていたロード時間が約20秒に短縮されますが、推論は非常に遅くなります。

✨ 主な機能

• 多言語対応：ace、acm、acqなど多数の言語に対応しています。
• 最先端のアルゴリズム：Expert Output Maskingを用いたトレーニングアルゴリズムを採用。
• 高速ロード：VRAM使用量を抑え、高速にモデルをロードできます。

📦 インストール

このモデルを使用するには、bitsandbytesを最新バージョンにアップグレードし、以下のhf transformersのプルリクエストをインストールします。

https://github.com/huggingface/transformers/pull/26037

💻 使用例

基本的な使用法

>>> from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

>>> tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-moe-54b")
>>> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-moe-54b")

>>> batched_input = [
'We now have 4-month-old mice that are non-diabetic that used to be diabetic," he added.',
"Dr. Ehud Ur, professor of medicine at Dalhousie University in Halifax, Nova Scotia and chair of the clinical and scientific division of the Canadian Diabetes Association cautioned that the research is still in its early days.",
"Like some other experts, he is skeptical about whether diabetes can be cured, noting that these findings have no relevance to people who already have Type 1 diabetes.",
"On Monday, Sara Danius, permanent secretary of the Nobel Committee for Literature at the Swedish Academy, publicly announced during a radio program on Sveriges Radio in Sweden the committee, unable to reach Bob Dylan directly about winning the 2016 Nobel Prize in Literature, had abandoned its efforts to reach him.",
'Danius said, "Right now we are doing nothing. I have called and sent emails to his closest collaborator and received very friendly replies. For now, that is certainly enough."',
"Previously, Ring's CEO, Jamie Siminoff, remarked the company started when his doorbell wasn't audible from his shop in his garage.",
]
>>> inputs = tokenizer(batched_input, return_tensors="pt", padding = True)

>>> translated_tokens = model.generate(
...     **inputs, forced_bos_token_id=tokenizer.lang_code_to_id["fra_Latn"]
... )
>>> tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)
['"Nous avons maintenant des souris de 4 mois non diabétiques qui étaient diabétiques", a-t-il ajouté.',
"Le docteur Ehud Ur, professeur de médecine à l'université Dalhousie, à Halifax, en Nouvelle-Écosse, et président de la division clinique et scientifique de l'Association canadienne du diabète, prévient que la recherche n'en est qu'à ses débuts.",
"Comme d'autres spécialistes, il est sceptique quant à la guérison du diabète, notant que ces résultats ne sont pas pertinents pour les personnes atteintes de diabète de type 1.",
"Lundi, Sara Danius, secrétaire permanente du Comité Nobel de littérature à l'Académie suédoise, a annoncé publiquement lors d'une émission de radio sur Sveriges Radio en Suède que le comité, incapable de contacter Bob Dylan directement au sujet du prix Nobel de littérature 2016, avait abandonné ses efforts pour le joindre.",
"Danius a déclaré: \"Pour le moment, nous ne faisons rien. J'ai appelé et envoyé des courriels à son plus proche collaborateur et j'ai reçu des réponses très amicales. Pour l'instant, c'est certainement suffisant\".",
"Auparavant, le PDG de Ring, Jamie Siminoff, a fait remarquer que la société avait commencé lorsque sa sonnette n'était pas audible depuis son magasin dans son garage.",
"Il a construit une sonnette WiFi, il a dit.",
]

📚 ドキュメント

トレーニング

• トレーニングにはExpert Output Maskingが使用されており、一部のトークンの全貢献をドロップすることで行われます。これは以下のスキームに対応しています。

生成

利用可能なチェックポイントには約350GBのストレージが必要です。マシンに十分なRAMがない場合は、accelerateを使用することを確認してください。

ターゲットテキストを生成する際には、forced_bos_token_idをターゲット言語IDに設定してください。以下の例は、facebook/nllb - moe - 54bモデルを使用して英語からフランス語に翻訳する方法を示しています。

なお、フランス語のBCP - 47コードfra_Latnを使用しています。Flores 200データセットのすべてのBCP - 47コードのリストはこちらを参照してください。

🔧 技術詳細

• トレーニングアルゴリズム：Expert Output Maskingを用いたトレーニングアルゴリズムが使用されています。
• 使用データセット：flores - 200データセットが使用されています。
• 評価指標：bleu、spbleu、chrf++が使用されています。

📄 ライセンス

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

言語情報