Falcon E 3B Base

🚀 transformers

このライブラリは、BitNetやFalcon - Eなどのモデルをサポートし、自然言語処理タスクを効率的に行うためのツールを提供します。

🚀 クイックスタート

このドキュメントでは、Falcon - Eシリーズのモデルについて、詳細な情報と使用方法を提供します。

✨ 主な機能

• 様々な言語モデルをサポートし、推論や微調整が可能です。
• 低メモリ使用量での高性能を実現します。

📦 インストール

Falcon - Eモデルを使用するには、Hugging FaceのtransformersライブラリまたはBitNetライブラリを利用できます。

💻 使用例

基本的な使用法

transformersを使用した推論

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "tiiuae/Falcon-E-1B-Base"

model = AutoModelForCausalLM.from_pretrained(
  model_id,
  torch_dtype=torch.bfloat16,
).to("cuda")

# Perform text generation

classic bfloat16 バージョンを使用した推論

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "tiiuae/Falcon-E-1B-Base"
revision = "bfloat16"

model = AutoModelForCausalLM.from_pretrained(
  model_id,
  torch_dtype=torch.bfloat16,
  revision=revision,
).to("cuda")

# Perform text generation

BitNetを使用した推論

git clone https://github.com/microsoft/BitNet && cd BitNet
pip install -r requirements.txt
python setup_env.py --hf-repo tiiuae/Falcon-E-1B-Base -q i2_s
python run_inference.py -m models/Falcon-E-1B-Base/ggml-model-i2_s.gguf -p "You are a helpful assistant" -cnv

高度な使用法

モデルの微調整

import torch

from transformers import AutoModelForCausalLM, AutoTokenizer
from trl import SFTTrainer
+ from onebitllms import replace_linear_with_bitnet_linear, quantize_to_1bit

model_id = "tiiuae/Falcon-E-1B-Base"

tokenizer = AutoTokenizer.from_pretrained(model_id, revision="prequantized")
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
+    revision="prequantized"
)
+ model = replace_linear_with_bitnet_linear(model)

trainer = SFTTrainer(
    model,
    ...
)

trainer.train()

+ quantize_to_1bit(output_directory)

📚 ドキュメント

モデルの詳細

学習の詳細

このモデルの学習プロトコルの詳細については、Falcon - E技術ブログを参照してください。

評価

以下の表は、内部パイプラインのベンチマーク結果です。評価結果は、以前のHugging Faceリーダーボードv2タスクの正規化されたスコアです。

以下は、命令微調整済みモデルの結果です。

有用なリンク

• リリースブログを参照してください。
• onebitllmsライブラリについて詳しく学ぶことができます。
• 質問がある場合や、研究者や開発者と交流したい場合は、Discordサーバーに参加してください。

引用

Falcon - Eシリーズのモデルがあなたの研究に役立った場合は、以下のように引用してください。

@misc{tiionebitllms,
    title = {Falcon-E, a series of powerful, universal and fine-tunable 1.58bit language models.},
    author = {Falcon-LLM Team},
    month = {April},
    url = {https://falcon-lm.github.io/blog/falcon-edge},
    year = {2025}
}

📄 ライセンス

このモデルは、Falcon - LLM Licenseの下で提供されています。詳細はこちらを参照してください。