🚀 Falcon-H1模型
Falcon-H1是一款由tiiuae开发的语言模型,采用混合Transformer与Mamba架构,支持英语及多语言。它在多种任务上表现出色,用户可通过Hugging Face transformers
、vLLM
或自定义的llama.cpp
库使用该模型。
🚀 快速开始
目前,若要使用此模型,你可以借助Hugging Face的transformers
、vLLM
库,或者我们自定义的llama.cpp
库的分支。
推理
确保安装transformers
或vLLM
的最新版本,必要时从源代码安装这些包:
pip install git+https://github.com/huggingface/transformers.git
更多关于从源代码构建vLLM的详细信息,请参考官方vLLM文档。
🤗 transformers
参考以下代码片段,使用🤗 transformers运行H1模型:
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "tiiuae/Falcon-H1-1B-Base"
model = AutoModelForCausalLM.from_pretrained(
model_id,
torch_dtype=torch.bfloat16,
device_map="auto"
)
vLLM
对于vLLM,只需执行以下命令启动服务器:
# pip install vllm
vllm serve tiiuae/Falcon-H1-1B-Instruct --tensor-parallel-size 2 --data-parallel-size 1
llama.cpp
我们正在将架构直接集成到llama.cpp
库中,在此期间,你可以安装我们的库分支并直接使用:https://github.com/tiiuae/llama.cpp-Falcon-H1,安装指南与llama.cpp
相同。
✨ 主要特性
- 模型类型:因果解码器
- 架构:混合Transformer + Mamba架构
- 支持语言:英语、多语言
- 许可证:Falcon-LLM许可证
📦 安装指南
安装transformers
pip install git+https://github.com/huggingface/transformers.git
安装vLLM
参考官方vLLM文档从源代码构建。
📚 详细文档
模型详情
- 开发者:https://www.tii.ae
- 模型类型:因果解码器
- 架构:混合Transformer + Mamba架构
- 支持语言:英语、多语言
- 许可证:Falcon-LLM许可证
训练详情
有关此模型训练协议的更多详细信息,请参考Falcon-H1技术博客文章。
评估
Falcon-H1系列在各种任务上表现出色,包括推理任务。具体评估结果如下:
任务 |
Falcon-H1-1.5B-deep |
Qwen3-1.7B |
Qwen2.5-1.5B |
Gemma3-1B |
Llama3.2-1B |
Falcon3-1B |
通用任务 |
|
|
|
|
|
|
BBH |
52.37 |
43.05 |
40.55 |
30.26 |
30.72 |
35.24 |
MMLU |
66.29 |
62.46 |
61.13 |
26.33 |
32.39 |
45.14 |
ARC-C |
55.89 |
55.72 |
54.27 |
39.33 |
39.42 |
47.87 |
HellaSwag |
69.72 |
67.09 |
67.86 |
62.94 |
65.73 |
62.3 |
Winogrande |
67.09 |
66.3 |
64.56 |
62.59 |
62.75 |
61.17 |
数学任务 |
|
|
|
|
|
|
GSM8k |
68.69 |
70.74 |
63.0 |
2.2 |
7.05 |
34.95 |
MATH lvl5 |
24.77 |
16.39 |
8.84 |
1.21 |
0.98 |
3.4 |
科学任务 |
|
|
|
|
|
|
GPQA |
32.8 |
29.45 |
28.36 |
24.66 |
23.57 |
27.85 |
MMLU-Pro |
41.07 |
33.81 |
28.72 |
11.31 |
11.8 |
16.11 |
MMLU-stem |
67.43 |
61.53 |
54.93 |
27.59 |
30.19 |
40.06 |
代码任务 |
|
|
|
|
|
|
HumanEval |
52.44 |
67.68 |
35.37 |
6.71 |
18.9 |
10.37 |
HumanEval+ |
46.34 |
60.98 |
29.27 |
5.49 |
16.46 |
9.15 |
MBPP |
70.9 |
67.72 |
60.05 |
12.7 |
35.98 |
12.43 |
MBPP+ |
60.32 |
58.99 |
49.47 |
9.52 |
29.89 |
9.52 |
更多详细评估信息可查看发布博客文章。
引用
如果Falcon-H1系列模型对你的工作有帮助,请引用:
@misc{tiifalconh1,
title = {Falcon-H1: A Family of Hybrid-Head Language Models Redefining Efficiency and Performance},
url = {https://falcon-lm.github.io/blog/falcon-h1},
author = {Falcon-LLM Team},
month = {May},
year = {2025}
}
📄 许可证
此模型使用Falcon-LLM许可证,详情请见https://falconllm.tii.ae/falcon-terms-and-conditions.html。
🔗 实用链接