library_name: transformers language:

• 阿拉伯语
• 捷克语
• 德语
• 英语
• 西班牙语
• 法语
• 印地语
• 意大利语
• 日语
• 韩语
• 荷兰语
• 波兰语
• 葡萄牙语
• 罗马尼亚语
• 俄语
• 瑞典语
• 乌尔都语
• 中文 tags:
• falcon-h1 license: other license_name: falcon-llm-license license_link: https://falconllm.tii.ae/falcon-terms-and-conditions.html

目录

0. 摘要
1. 模型详情
2. 训练详情
3. 使用方法
4. 评估
5. 引用

摘要

模型详情

模型描述

• 开发机构: https://www.tii.ae
• 模型类型: 因果仅解码器架构
• 架构: 混合Transformer + Mamba架构
• 支持语言: 英语、多语种
• 许可证: Falcon-LLM许可证

训练详情

有关本模型训练协议的更多细节，请参阅Falcon-H1技术博客。

使用方法

当前可使用Hugging Face的transformers、vLLM或我们定制的llama.cpp分支库来调用该模型。

推理

确保安装最新版transformers或vllm，必要时从源码安装：

pip install git+https://github.com/huggingface/transformers.git

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相同。

评估

Falcon-H1系列在各类任务（包括推理任务）中表现优异。

详细基准测试请参阅发布博客。

实用链接

• 查看发布博客
• 加入Discord社区与研发团队交流

引用

若Falcon-H1系列模型对您的研究有所帮助，欢迎引用：

@misc{tiifalconh1,
    title = {Falcon-H1：重新定义效率与性能的混合头语言模型家族},
    url = {https://falcon-lm.github.io/blog/falcon-h1},
    author = {Falcon-LLM团队},
    month = {5月},
    year = {2025}
}