Tencent-Hunyuan-Large

GitHub
1.6k 122 困难 1 次阅读 2天前NOASSERTION语言模型
AI 解读 由 AI 自动生成,仅供参考

Tencent-Hunyuan-Large 是腾讯开源的一款超大规模语言模型,也是目前行业内最大的基于 Transformer 架构的混合专家(MoE)模型。它旨在解决大模型在规模扩张过程中,如何平衡卓越性能与资源消耗的行业难题。

该模型拥有惊人的 3890 亿总参数量,但在实际推理时仅激活 520 亿参数。这种独特的“混合专家”机制使其能够在保持顶尖任务处理能力的同时,显著降低计算成本和延迟,实现了高效能与低资源的完美统一。无论是复杂的自然语言理解、逻辑推理,还是长文本处理,它都能提供出色的表现。

Tencent-Hunyuan-Large 特别适合 AI 研究人员、大模型开发者以及企业技术团队使用。研究人员可借此深入探索 MoE 架构的前沿技术;开发者则能利用其开源权重(包括 FP8 量化版本),在相对有限的算力条件下部署高性能应用,构建智能客服、内容创作或数据分析系统。通过开放技术细节与模型权重,Tencent-Hunyuan-Large 致力于推动社区创新,让超大模型的强大能力更易于被获取和应用。

使用场景

某大型金融科技公司正在构建新一代智能投研系统,需要处理海量长篇财报、研报及实时新闻,以生成深度市场分析摘要。

没有 Tencent-Hunyuan-Large 时

  • 长文档理解能力不足:现有模型上下文窗口有限,无法一次性读完百页级的完整财报,导致关键信息遗漏或分析碎片化。
  • 推理成本高昂:为了覆盖复杂逻辑,团队被迫调用多个专用小模型串联,不仅维护成本高,且整体响应延迟严重。
  • 专业深度欠缺:通用模型在金融术语理解和复杂因果推导上表现平平,生成的报告常出现事实性幻觉,需人工大量复核。
  • 资源利用率低:传统稠密模型在处理简单查询时也全量激活参数,造成算力浪费,难以支撑高并发场景。

使用 Tencent-Hunyuan-Large 后

  • 超长上下文精准掌控:依托其 256k 的超长上下文窗口,Tencent-Hunyuan-Large 能一次性“吃透”整本财报,确保摘要逻辑连贯、数据准确无误。
  • MoE 架构降本增效:作为拥有 3890 亿总参数但仅激活 520 亿的混合专家(MoE)模型,Tencent-Hunyuan-Large 在保持顶尖性能的同时,大幅降低了单次推理的算力消耗。
  • 行业洞察更深刻:得益于超大规模参数的知识储备,Tencent-Hunyuan-Large 能精准识别金融隐喻与复杂市场关联,输出的分析结论具备专家级深度。
  • 动态路由提升速度:针对不同难度的任务,Tencent-Hunyuan-Large 自动调度相应专家网络,显著提升了系统在高并发下的响应速度与稳定性。

Tencent-Hunyuan-Large 通过业界最大的开源 MoE 架构,完美平衡了超长文本处理能力与推理成本,让企业级深度智能分析变得触手可及。

运行环境要求

操作系统
  • 未说明
GPU
  • 必需 NVIDIA GPU
  • 支持 FP8 量化(显存占用比 FP16/BF16 减少 50%)和全精度运行
  • 推荐使用支持 TensorRT-LLM 或 vLLM 后端的显卡以利用 CLA 结构优化(KV Cache 显存节省 50%)
  • 具体显存大小取决于模型版本(FP8 或 FP16)及上下文长度(最高 256K),未给出具体最低显存数值,但鉴于模型规模(389B 总参数/52B 激活参数),需多卡或多节点环境
内存

未说明

依赖
notes该模型为业界最大的开源 MoE 模型(总参数 3890 亿,激活参数 520 亿)。提供 FP8 量化版本以大幅降低显存需求并提升吞吐量。推理后端支持定制的 vLLM(已开源)和 TensorRT-LLM(即将开源),其中 TRT-LLM 性能优于 vLLM 30% 以上。训练兼容 Hugging Face 格式,可使用 hf-deepspeed 框架进行微调。支持超长上下文(预训练 256K,指令版 128K)。
python未说明
vLLM (定制分支)
TensorRT-LLM (计划开源)
Hugging Face Transformers
DeepSpeed (hf-deepspeed)
Flash Attention
Tencent-Hunyuan-Large hero image

快速开始

中文  | 英文




🫣 Hugging Face   |   🖥️  官方网站  |  🕖·  HunyuanAPI  |  🐳·  Gitee

技术报告  |   演示   |   Tencent Cloud TI   


模型下载
模型 Hugging Face 下载链接 腾讯云下载链接
Hunyuan-A52B-Instruct-FP8 Hunyuan-A52B-Instruct-FP8 Hunyuan-A52B-Instruct-FP8
Hunyuan-A52B-Instruct Hunyuan-A52B-Instruct Hunyuan-A52B-Instruct
Hunyuan-A52B-Pretrain Hunyuan-A52B-Pretrain Hunyuan-A52B-Pretrain

模型介绍

随着人工智能技术的快速发展,大型语言模型(LLMs)在自然语言处理、计算机视觉以及科学任务等领域取得了显著进展。然而,随着模型规模的不断扩大,如何在保持高性能的同时优化资源消耗已成为关键挑战。为应对这一挑战,我们探索了专家混合(MoE)模型。目前发布的 Hunyuan-Large(Hunyuan-MoE-A52B)模型是业界最大的开源 Transformer 基础 MoE 模型,总参数量达 3890 亿,活跃参数量为 520 亿。这是目前业界最大的开源 Transformer 基础 MoE 模型,总参数量达 3890 亿,活跃参数量为 520 亿。

通过开源 Hunyuan-Large 模型并公开相关技术细节,我们希望激发更多研究人员的创新灵感,共同推动人工智能技术的进步与应用。我们诚挚邀请您加入我们的开源社区,一起探索和优化未来的 AI 模型!

技术优势介绍

模型

  • 高质量合成数据:通过引入合成数据进行训练,Hunyuan-Large 能够学习到更丰富的表征,更好地处理长上下文输入,并对未见数据具有更强的泛化能力。

  • KV 缓存压缩:采用分组查询注意力(GQA)和跨层注意力(CLA)策略,显著降低 KV 缓存的内存占用和计算开销,从而提升推理吞吐量。

  • 专家特定学习率缩放:为不同专家设置不同的学习率,确保每个子模型都能有效从数据中学习,并为整体性能做出贡献。

  • 长上下文处理能力:预训练模型支持高达 256K 的文本序列,而 Instruct 模型则支持高达 128K,极大地提升了处理长上下文任务的能力。

  • 广泛的基准测试:我们在多种语言和任务上进行了大量实验,以验证 Hunyuan-Large 的实际效果和安全性。

推理框架

  • 此次开源提供了两种针对 Hunyuan-Large 模型的推理后端选项:流行的 vLLM 后端 和 TensorRT-LLM 后端。这两种方案都包含了性能优化措施。例如,引入新的 CLA 结构显著减少了 GPU 内存的使用,使 KV 缓存部分节省了 50%,从而确保高效处理长文本场景。此外,通过采用 FP8 量化,相比传统的 FP16/BF16 量化,内存占用减少了 50%,同时保持了精度,并使吞吐量提高了 70%。与此同时,借助 TRT-LLM 核心的高效算子,TRT-LLM 方案的性能比 vLLM 高出 30% 以上。TRT-LLM 方案已在腾讯的 Hunyuan 项目中广泛应用。本次发布我们首先开源的是 vLLM 方案,计划在不久的将来推出 TRT-LLM 方案。

训练框架

  • Hunyuan-Large 开源模型完全兼容 Hugging Face 格式,研究人员和开发者可以使用 hf-deepspeed 框架进行模型微调。此外,我们还支持通过 Flash Attention 加速训练。为进一步帮助大家快速上手,我们通过此次发布将相应的训练脚本和模型实现公开给社区,方便后续基于这些资源进行模型训练和微调操作。

 

相关新闻

  • 2024年11月25日 我们自主研发的长文本基准评测集——PenguinScrolls正式发布!您可以在GitHub上探索该项目,并在Hugging Face上获取数据集。
  • 2024年11月18日 Hunyuan-A52B-InstructHunyuan-A52B-Instruct-FP8 模型更新。
  • 2024年11月5日 TI 平台 已集成 Hunyuan-Large 模型,您只需简单几步即可轻松进行训练和部署。请访问与 Hunyuan-Large 聊天,体验与该模型的实时对话;同时,您还可以参阅TI 上的 Hunyuan-Large 最佳实践,打造属于您的定制化 Hunyuan-Large 模型。
  • 2024年11月5日 我们已在 Hugging Face 上开源了 Hunyuan-A52B-PretrainHunyuan-A52B-InstructHunyuan-A52B-Instruct-FP8。此外,我们还发布了技术报告以及训练与推理操作手册,详细介绍了模型的能力及训练和推理的具体流程。

基准评估

Hunyuan-Large 预训练模型 在与激活参数规模相近的密集型和 MoE 架构竞争对手相比,整体性能表现最佳。在 MMLU、MMLU-Pro 和 CMMLU 等综合基准测试中,Hunyuan-Large 始终取得最优成绩,充分证明了其在综合性任务上的全面能力。

此外,Hunyuan-Large 在常识理解与推理,以及经典 NLP 任务(如问答和阅读理解)方面也展现出卓越性能,例如 CommonsenseQA、PIQA 和 TriviaQA 等数据集。在数学能力方面,Hunyuan-Large 在 GSM8K 和 MATH 数据集上均超越所有基线模型,并且在中国语种的 CMATH 数据集中也取得了最佳成绩。同时,在所有中文任务中(如 CMMLU 和 C-Eval),Hunyuan-Large 同样表现出整体最优性能。

模型 LLama3.1-405B LLama3.1-70B Mixtral-8x22B DeepSeek-V2 Hunyuan-Large
MMLU 85.2 79.3 77.8 78.5 88.4
MMLU-Pro 61.6 53.8 49.5 - 60.2
BBH 85.9 81.6 78.9 78.9 86.3
HellaSwag - - 88.7 87.8 86.8
CommonsenseQA 85.8 84.1 82.4 - 92.9
WinoGrande 86.7 85.3 85.0 84.9 88.7
PIQA - - 83.6 83.7 88.3
NaturalQuestions - - 39.6 38.7 52.8
DROP 84.8 79.6 80.4 80.1 88.9
ARC-C 96.1 92.9 91.2 92.4 95.0
TriviaQA - - 82.1 79.9 89.2
CMMLU - - 60.0 84.0 90.2
C-Eval - - 59.6 81.7 91.9
C3 - - 71.4 77.4 82.3
GSM8K 89.0 83.7 83.7 79.2 92.8
MATH 53.8 41.4 42.5 43.6 69.8
CMATH - - 72.3 78.7 91.3
HumanEval 61.0 58.5 53.1 48.8 71.4
MBPP 73.4 68.6 64.2 66.6 72.6

Hunyuan-Large-Instruct 相较于具有相似激活参数量的 LLM,在大多数类型的任务上都实现了持续提升,这表明我们的后训练方法非常有效。深入分析不同类别基准测试中的模型表现,我们发现该指令微调模型在 MMLU 和 MATH 数据集上取得了最佳成绩。尤其值得一提的是,在 MMLU 数据集中,我们的模型表现显著提升,比 LLama3.1-405B 模型高出 2.6%。这一进步并非微小,而是充分体现了 Hunyuan-Large-Instruct 在广泛的语言理解任务中所具备的卓越理解和推理能力。此外,该模型在 MATH 数据集上的表现同样令人瞩目,以 3.6% 的优势领先 LLama3.1-405B。值得注意的是,如此大幅度的准确率提升仅依靠 520 亿个激活参数便得以实现,进一步凸显了我们模型的高效性。

模型 LLama3.1 405B Inst. LLama3.1 70B Inst. Mixtral 8x22B Inst. DeepSeekV2.5 Chat Hunyuan-Large Inst.
MMLU 87.3 83.6 77.8 80.4 89.9
CMMLU - - 61.0 - 90.4
C-Eval - - 60.0 - 88.6
BBH - - 78.4 84.3 89.5
HellaSwag - - 86.0 90.3 88.5
ARC-C 96.9 94.8 90.0 - 94.6
GPQA_diamond 51.1 46.7 - - 42.4
MATH 73.8 68.0 49.8 74.7 77.4
HumanEval 89.0 80.5 75.0 89.0 90.0
AlignBench 6.0 5.9 6.2 8.0 8.3
MT-Bench 9.1 8.8 8.1 9.0 9.4
IFEval strict-prompt 86.0 83.6 71.2 - 85.0
Arena-Hard 69.3 55.7 - 76.2 81.8
AlpacaEval-2.0 39.3 34.3 30.9 50.5 51.8

 

数据

Hunyuan-Large 提供了与模型训练相关的流程。在本节中,您可以对训练数据格式进行处理,以便用于模型训练。

训练数据格式及处理

训练数据被处理为如下消息格式。训练和推理时的默认系统提示词是“你是一个乐于助人的助手”。以下是单轮和多轮数据的示例:

# 单轮数据格式

messages = [
    {"role": "system", "content": "你是一个乐于助人的助手。"},
    {"role": "user", "content": "海水为什么是咸的" },
    {"role": "assistant", "content": "海水是咸的主要是因为其中含有许多溶解在水中的盐类和矿物质。这些盐类和矿物质来自于地球表面的岩石和土壤中的化学物质,随着时间的推移,它们被带到了海洋中。当海水蒸发时,水分蒸发掉了,但盐类和矿物质仍然留在水中,导致海水变得更加咸味。因此,海水的咸度是由其中的盐类和矿物质的含量决定的。"}
]

# 多轮对话数据格式
messages = [
    {"role": "system", "content": "你是一个 helpful assistant。"},
    {"role": "user", "content": "知道重庆森林这部电影吗?" },
    {"role": "assistant", "content": "知道呀,是一部由王家卫导演的片子。" },
    {"role": "user", "content": " 而主演里更是有王菲,一上映便受到追捧。" },
    {"role": "assistant", "content": "所以此片获得了第14届香港电影金像奖最佳影片奖。"}
]

from tokenizers import ByteLevelBPETokenizer
from transformers import AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("./models", use_fast=False, trust_remote_code=True)

ids = tokenizer.apply_chat_template(messages)

更多使用参考,请参阅 ./models/test.py 文件。

 

快速入门

您可以通过参考快速入门指南中的内容,快速上手。

模型训练

为了简化训练流程,HunyuanLLM 提供了一个预构建的 Docker 镜像:

hunyuaninfer/hunyuan-large

硬件要求

已在 H20 上进行测试,在未启用 make_moe_param_leaf_module 并使用 zero3+offload 的情况下,当 max_seq_length 设置为 2048 时,全量微调至少需要 32 张 GPU 卡,而 LoRA 微调则至少需要 8 张 GPU 卡。

训练性能

在最低配置下(8 张 GPU 卡用于 LoRA 微调),将 per_device_train_batch_size 设置为 1,同时将 gradient_accumulation_steps 设置为 1,每轮迭代大约需要 35 秒。

启动方法

请参考:HuggingFace Transformers Trainer

单机训练

train 目录下,执行以下命令:

pip install -r requirements.txt
bash train.sh

多机训练

要在多台机器上启动训练,请按照以下步骤操作,并确保所有机器位于同一个集群内。

配置机器间的无密码 SSH 登录

以下步骤以两台机器为例,其 IP 地址分别表示为 ${ip1}${ip2}。这些操作均在 Docker 容器中进行。

首先,在每台机器上的容器之间配置无密码 SSH 连接。

ssh-keygen			# 生成用于无密码登录的 id_rsa 和 id_rsa.pub
ssh-keygen -t rsa -A    # 生成 /etc/ssh/ssh_host_rsa_key 和 ssh_host_ecdsa_key,以便后续启动 'SSH listen'
/usr/sbin/sshd -p 36005 -o ListenAddress=0.0.0.0        # 启动 SSH 监听
echo "Port 36005" > ~/.ssh/config   # 将 SSH 连接端口更改为 36005
passwd root    # 设置 root 密码,以避免监控平台发出警报

注意:这里的 36005 是一个示例。您可以选择任意端口,但务必确保该端口已开放未被其他进程占用

接下来,在每台机器的容器内执行:

cat ~/.ssh/id_rsa.pub

复制输出的 SSH 公钥,并将其粘贴到 ~/.ssh/authorized_keys 文件中,每行放置一个公钥。这一步骤必须在每一台机器上完成。 最终,每台机器上的 ~/.ssh/authorized_keys 文件应完全相同,包含所有机器的公钥。

需要注意的是,在多节点训练过程中,每个节点上运行的代码必须保持一致。建议挂载共享网络驱动器。如果无法挂载共享驱动器,则需要手动将数据集、脚本和代码复制到所有机器的同一目录下。

开始多机训练

当准备工作完成后,并确认依赖项已安装(若未安装,请执行 pip install -r requirements.txt 进行安装)时,可以在 train.sh 文件的开头添加以下配置:

export HOST_GPU_NUM=8
# 当前机器 IP
export LOCAL_IP=${ip1}
# 多节点机器 IP 列表,用逗号分隔
export NODE_IP_LIST="${ip1}:8,${ip2}:8"

# 机器节点数量
export NODES=2
export NODE_NUM=$((${NODES} * ${HOST_GPU_NUM}))

注意:请将 ${ip1}${ip2} 替换为实际的 IP 地址!

然后,在 ${ip1} 所在的机器上,进入 train/ 目录并执行 bash train.sh。请注意,首次运行时,您可能会看到如下提示:

无法确认主机 '[ip]:36005 ([ip]:36005)' 的真实性。
ECDSA 密钥指纹是 xxxxxx。
ECDSA 密钥指纹的 MD5 哈希值是 xxxxxx。
您确定要继续连接吗(是/否)?

此时,请输入 yes 继续。

关键参数

脚本中的关键参数如下:

  • --deepspeed:该参数应指向 DeepSpeed 配置文件。train 文件夹提供了三个默认的 DeepSpeed 配置文件:ds_zero2_no_offload.jsonds_zero3_no_offload.jsonds_zero3_offload.json。所需显存依次减少。
  • --model_name_or_path:HF 预训练模型的路径。请确保该路径包含 modeling_hunyuan.pyconfiguration_hunyuan.py 文件,否则无法加载。
  • --tokenizer_name_or_path:分词器文件夹的路径。请确保该路径包含 tokenization_hy.py 文件,否则无法加载。
  • --train_data_file:训练数据文件的路径,应为 JSONL 格式。
  • --output_dir:输出目录,用于存储日志、TensorBoard 文件和模型权重。
  • --per_device_train_batch_size:每个 GPU 的批量大小。
  • --gradient_accumulation_steps:梯度累积步数。全局批量大小为 per_device_train_batch_size * gradient_accumulation_steps * dp_size
  • --max_steps:总训练步数。
  • --save_steps:保存检查点的间隔步数。
  • --use_lora:是否使用 LoRA 进行训练。此参数还接受 --lora_rank--lora_alpha--lora_dropout 参数。LoRA 默认应用于 'q_proj''k_proj''v_proj''o_proj' 参数。如需更改,请在代码中修改。注意:使用 LoRA 训练时,仅保存 LoRA 权重,不保存基础模型权重。若需合并 LoRA 权重,请参阅下方的“LoRA 权重合并”部分。
  • --make_moe_param_leaf_module:在使用 zero3 和 MoE 训练时,将 MoE 模块视为叶模块,即其参数不会被 zero3 切分。此选项预计将显著增加显存占用。
  • --gradient_checkpointing:启用梯度检查点。
  • --train_attention_params_only:是否仅训练注意力参数。
  • --learning_rate:训练过程中的最大学习率。
  • --min_lr:训练过程中的最小学习率。
  • --use_flash_attn:开启 flash-attention 以加速训练。

注意:

  • 如果您希望从先前保存的检查点继续训练,而不是加载预训练权重,请指定 --resume_from_checkpoint 并提供上一次训练的检查点路径。请勿同时指定 --model_name_or_path,因为那样只会加载权重,而不会恢复训练状态。
  • 从检查点继续训练时,由于某些非确定性算法引入的随机性,损失值可能会出现轻微偏差,这属于正常现象。详情请参阅:HuggingFace Transformers Trainer Randomness
  • 当指定了 --model_name_or_path 时,所有与模型相关的参数将被忽略。
  • 同一批次内的样本会被填充至与该批次中最长样本一致的长度,每个样本的最大长度为 max_seq_length。超出部分将被截断。
  • 如果遇到关于偏置权重未加载的警告,可以忽略,因为 Hunyuan-Large 中并未使用偏置项。

显存不足怎么办?

请参考:DeepSpeed 配置文档

您可以尝试修改 DeepSpeed 配置文件,移除这些参数的 auto 属性,并降低其数值:

  • stage3_param_persistence_threshold
  • stage3_prefetch_bucket_size
  • stage3_max_reuse_distance
  • stage3_max_reuse_distance

LoRA 模型的合并

在启用 zero3 的情况下,由于模型权重被分割到不同的数据并行 Rank 上,因此无法在训练过程中直接将 LoRA 权重合并到零3模型中。如果需要将 LoRA 权重合并到基础模型中,可以在离线状态下完成合并操作,生成合并后的权重文件。执行 merge_lora_weight.sh 脚本即可将 LoRA 权重与基础模型权重合并。该脚本的参数包括:

  • --base_model_path:基础模型权重的目录。
  • --adapter_model_path:LoRA 权重的目录。
  • --output_path:保存合并后权重的目录。
  • --save_dtype:合并后权重的存储数据类型,可选格式有:fp16、bf16、fp32。

 

推理与部署

HunyuanLLM 使用 TRT-LLM 和 vLLM 进行部署。我们已开源 vLLM-backend 部署方案(详见“使用 vLLM 进行推理”),而 TRT-LLM 部署方案(详见“使用 TRT-LLM 进行推理”)将在不久的将来发布。

使用 TRT-LLM 进行推理

待开放

使用 vLLM 进行推理

Docker:

为简化部署流程,HunyuanLLM 提供了一个预构建的 Docker 镜像:

hunyuaninfer/hunyuan-large。您只需下载模型文件,并使用以下命令启动 Docker 容器,即可开始模型推理。

docker run --name hunyuanLLM_infer -itd --privileged --user root --net=host --ipc=host --gpus=8 hunyuaninfer/hunyuan-large:infer-open-source

注意:Docker 容器权限管理。上述命令以特权模式(--privileged)启动容器,这会赋予容器更高的权限,从而增加数据泄露和集群安全风险。建议尽量避免使用特权模式,以降低安全风险。对于必须使用特权模式的场景,请进行全面的安全评估,并采取适当的安全监控和加固措施。

配置多机之间的无密码 SSH 登录

以下步骤以两台机器为例,其 IP 分别表示为 ${ip1}${ip2}。这些操作均在 Docker 容器内进行。

首先,在两台机器上分别运行 passwd 设置密码,例如:Tmp123,./

inference/login_ssh.py 复制到容器中,并执行以下命令,确保正确输入 IP 和密码。

python3 login_ssh.py --ips ${ip1},${ip2} --port 36000 --password=Tmp123,./

注意 📢:在开始之前,请务必使用 vLLM 的调试脚本验证多机通信是否正常:https://docs.vllm.ai/en/latest/getting_started/debugging.html

BF16 部署

BF16 部署需要 16 块 H20 GPU。在确认多机通信正常后,执行以下步骤:

在运行命令之前,请设置以下环境变量:

${LOCAL_IP}:当前机器上 bond1 对应的 IP 地址
${MODEL_PATH}:Hunyuan LLM 模型路径

第一步:启动 Ray

Ray 是一个用于并行和分布式 Python 的开源库。在本节中,我们使用 Ray 实现多机通信。

Ray 组件配置加固:Ray 组件的默认配置未启用服务端口(例如 6379、8265)的身份验证机制,存在未授权访问和命令执行的风险。建议仅在受信任的内部网络环境中部署 Ray 组件,或确保对这些端口实施严格的访问控制列表(ACL)策略,以防止未经授权的网络访问。

首先,在每个节点上启动 Ray(可以在后台运行,也可以保持终端打开):

在主节点上:

export VLLM_HOST_IP=${LOCAL_IP}
export NCCL_SOCKET_IFNAME=bond1
export GLOO_SOCKET_IFNAME=bond1
ray start --block --head --node-ip-address=${LOCAL_IP} --port=6379

在所有工作节点上:

注意:将 {HEAD NODE $LOCAL_IP} 替换为主节点的实际 ${LOCAL_IP}。

export VLLM_HOST_IP=${LOCAL_IP}
export NCCL_SOCKET_IFNAME=bond1
export GLOO_SOCKET_IFNAME=bond1
ray start --block --address={HEAD NODE $LOCAL_IP}:6379 --node-ip-address=${LOCAL_IP}

如果 Ray 启动失败,请先执行 ray stop,然后再重新运行上述命令。

第二步:执行推理

方法一:命令行推理

以下代码片段展示了如何使用 vLLM 快速请求聊天模型:

注意:vLLM 组件远程代码执行防护。在下面的代码中,如果启用了 vLLM 组件的 trust-remote-code 配置选项,将允许从远程模型仓库加载并执行代码,这可能导致恶意代码被执行。除非业务明确要求,建议保持该配置选项关闭,以降低潜在的安全威胁。

import os
from vllm import LLM, SamplingParams

model_path=os.environ.get('MODEL_PATH')

llm = LLM(model=model_path,
        tokenizer=model_path,
        trust_remote_code=True,
        max_model_len=10240,
        dtype='bfloat16',
        tensor_parallel_size=16,
        pipeline_parallel_size=1,
        disable_log_stats=False,
        gpu_memory_utilization=0.98,
        disable_custom_all_reduce=True,
        #distributed_executor_backend='ray',
        enforce_eager=True,
        max_num_seqs=8,
        use_v2_block_manager=True,
        quantization=None)

prompts = ["海水为什么是咸的"]

sampling_params = SamplingParams(
    temperature=0.7, top_p=0.6, max_tokens=200, top_k=20, repetition_penalty=1.05)

outputs = llm.generate(prompts, sampling_params)

# 打印输出结果。
for output in outputs:
    prompt = output.prompt
    generated_text = output.outputs[0].text
    print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")

方法二:基于服务的推理

下面我们演示如何使用 vLLM 以服务方式部署模型并进行请求。

在主节点上运行以下命令:

export VLLM_HOST_IP=${LOCAL_IP}
export NCCL_SOCKET_IFNAME=bond1
export GLOO_SOCKET_IFNAME=bond1

然后通过运行以下命令启动服务:

cd inference
sh run_server.sh

提示:故障排除,如果遇到以下错误:

ray.exceptions.RaySystemError: System error: No module named 'transformers_modules' traceback: Traceback (most recent call last):
ModuleNotFoundError: No module named 'transformers_modules'

请将主节点上的 ~/.cache/huggingface/modules/ 目录复制到所有工作节点的相应路径。

成功运行 run_server.sh 后,执行请求脚本:

sh openapi.sh

务必修改 openapi.sh 中的 ${LOCAL_IP}${MODEL_PATH},使其与对应服务的值一致。

量化模型部署:

本节介绍使用 vLLM 部署量化模型的过程。

图片:部署镜像与 BF16 相同。

Int8 量化模型部署:

要部署 Hunyuan-L 模型的 Int8 权重专用版本,只需在 run_server_int8.sh 中设置以下环境变量:

${MODEL_PATH}:BF16 模型路径
${LOCAL_IP}:当前机器上 bond1 对应的 IP 地址

然后通过运行以下命令启动 Int8 服务:

sh run_server_int8.sh

成功运行 run_server_int8.sh 后,执行请求脚本:

sh openapi.sh

FP8 量化模型部署:

要部署 Hunyuan-L 模型的 W8A8C8 版本,只需在 run_server_fp8.sh 中设置以下环境变量:

${MODEL_PATH}:FP8 模型路径
${LOCAL_IP}:当前机器上 bond1 对应的 IP 地址

然后通过运行以下命令启动 FP8 服务:

sh run_server_fp8.sh

成功运行 run_server_fp8.sh 后,执行请求脚本:

sh openapi.sh

FP8 基准测试

本部分介绍 Hunyuan Large Instruct FP8 量化模型的基准测试结果。

数据集 BF16 W8A8C8-FP8
ARC-C 94.6 94.2
C-Eval 88.6 89.2
CMMLU 90.4 89.8
MMLU 89.9 88.9

推理性能

本节展示了使用 vLLM 部署各种模型(原始模型及量化模型)的效率测试结果,包括不同批处理大小下的推理速度(tokens/s)。

推理框架 模型 GPU 数量(H20) input_length batch=1 batch=4
vLLM Hunyuan-Large 16 2048 20.2 75.5
vLLM Hunyuan-Large(仅权重为 Int8) 8 2048 19.3 73.6
vLLM Hunyuan-Large(W8A8C8-FP8) 8 2048 19.8 74.9

分词器

HunYuan-Large 模型所使用的分词器在压缩率和效果之间取得了平衡,确保嵌入向量能够得到充分训练。该分词器的词汇表包含 10 万个来自 tiktoken 的词元,并在此基础上利用大量高质量的中文训练数据额外训练了 2.9 万个中文词元,以增强模型的中文能力及分词器的压缩率。综合来看,我们的新分词器相比 LLaMA3 的分词器,压缩率从每词元 2.78 个字符提升至 3.13 个字符。

Hunyuan API

您可以在腾讯云上体验我们的 Hunyuan-Large 模型。详情请访问:https://cloud.tencent.com/document/product/1729/97730。

交互式 Demo 网页

Hunyuan-Large 的网页版 Demo 已正式开放。请访问 https://huggingface.co/spaces/tencent/Hunyuan-Large,轻松体验我们的模型。

在 TI 平台上进行训练与推理

腾讯云的 TI 平台 是专为 AI 工程师打造的综合性机器学习平台。Hunyuan-Large 模型已集成其中,您只需简单几步即可完成训练与部署。访问 与 Hunyuan-Large 对话,实时体验与模型的对话;同时,您还可以参阅 Hunyuan-Large 在 TI 上的最佳实践,创建属于您自己的定制化 Hunyuan-Large 模型。

引用

如果您觉得我们的工作对您有所帮助,请随时引用我们。

@misc{sun2024hunyuanlargeopensourcemoemodel,
      title={Hunyuan-Large: 腾讯开源的拥有 520 亿激活参数的 MoE 模型}, 
      author={孙兴武、陈彦峰、黄一青、谢若冰、朱嘉琪、张凯、李帅鹏、杨震、韩约翰、舒晓波、卜家豪、陈中志、黄雪萌、连奉宗、杨赛勇、严建峰、曾宇远、任小琴、余超、吴露露、毛悦、杨涛、郑孙聪、吴侃、焦典、薛金宝、张希鹏、吴德成、刘凯、吴登鹏、徐广辉、陈绍华、陈爽、冯晓、洪义庚、郑俊强、许成成、李宗伟、匡雄、胡江陆、陈怡琪、邓雨驰、李贵阳、刘傲、张晨晨、胡世辉、赵子龙、吴子凡、丁耀、王伟超、刘汉、罗伯茨·王、费浩、佘佩杰、赵泽、曹勋、王海、项福生、黄梦源、熊志远、胡斌、侯学斌、姜磊、吴佳佳、邓亚平、沈毅、王倩、刘伟杰、刘杰、陈猛、董亮、贾卫文、陈虎、刘飞飞、袁睿、徐慧琳、闫振祥、曹腾飞、胡志超、冯新华、杜东、佘廷昊、陶阳宇、张峰、朱建臣、许成忠、李西瑞、查重、欧阳文、夏寅本、李翔、何泽坤、陈荣鹏、宋嘉伟、陈瑞彬、蒋凡、赵重庆、王博、龚浩、甘荣、胡温斯顿、康展辉、杨勇、刘玉红、王迪、江杰},
      year={2024},
      eprint={2411.02265},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2411.02265}, 
}

联系我们

如果您想向我们的研发和产品团队留言,欢迎联系我们的开源团队。您也可以通过电子邮件(hunyuan_opensource@tencent.com)与我们取得联系。

常见问题

相似工具推荐

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

155.4k|★★☆☆☆|今天
开发框架Agent语言模型

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|1周前
语言模型图像Agent

NextChat

NextChat 是一款轻量且极速的 AI 助手,旨在为用户提供流畅、跨平台的大模型交互体验。它完美解决了用户在多设备间切换时难以保持对话连续性,以及面对众多 AI 模型不知如何统一管理的痛点。无论是日常办公、学习辅助还是创意激发,NextChat 都能让用户随时随地通过网页、iOS、Android、Windows、MacOS 或 Linux 端无缝接入智能服务。 这款工具非常适合普通用户、学生、职场人士以及需要私有化部署的企业团队使用。对于开发者而言,它也提供了便捷的自托管方案,支持一键部署到 Vercel 或 Zeabur 等平台。 NextChat 的核心亮点在于其广泛的模型兼容性,原生支持 Claude、DeepSeek、GPT-4 及 Gemini Pro 等主流大模型,让用户在一个界面即可自由切换不同 AI 能力。此外,它还率先支持 MCP(Model Context Protocol)协议,增强了上下文处理能力。针对企业用户,NextChat 提供专业版解决方案,具备品牌定制、细粒度权限控制、内部知识库整合及安全审计等功能,满足公司对数据隐私和个性化管理的高标准要求。

87.6k|★★☆☆☆|1周前
开发框架语言模型

ML-For-Beginners

ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。

85.1k|★★☆☆☆|4天前
图像数据工具视频

funNLP

funNLP 是一个专为中文自然语言处理(NLP)打造的超级资源库,被誉为"NLP 民工的乐园”。它并非单一的软件工具,而是一个汇集了海量开源项目、数据集、预训练模型和实用代码的综合性平台。 面对中文 NLP 领域资源分散、入门门槛高以及特定场景数据匮乏的痛点,funNLP 提供了“一站式”解决方案。这里不仅涵盖了分词、命名实体识别、情感分析、文本摘要等基础任务的标准工具,还独特地收录了丰富的垂直领域资源,如法律、医疗、金融行业的专用词库与数据集,甚至包含古诗词生成、歌词创作等趣味应用。其核心亮点在于极高的全面性与实用性,从基础的字典词典到前沿的 BERT、GPT-2 模型代码,再到高质量的标注数据和竞赛方案,应有尽有。 无论是刚刚踏入 NLP 领域的学生、需要快速验证想法的算法工程师,还是从事人工智能研究的学者,都能在这里找到急需的“武器弹药”。对于开发者而言,它能大幅减少寻找数据和复现模型的时间;对于研究者,它提供了丰富的基准测试资源和前沿技术参考。funNLP 以开放共享的精神,极大地降低了中文自然语言处理的开发与研究成本,是中文 AI 社区不可或缺的宝藏仓库。

79.9k|★☆☆☆☆|6天前
语言模型数据工具其他

gpt4all

GPT4All 是一款让普通电脑也能轻松运行大型语言模型(LLM)的开源工具。它的核心目标是打破算力壁垒,让用户无需依赖昂贵的显卡(GPU)或云端 API,即可在普通的笔记本电脑和台式机上私密、离线地部署和使用大模型。 对于担心数据隐私、希望完全掌控本地数据的企业用户、研究人员以及技术爱好者来说,GPT4All 提供了理想的解决方案。它解决了传统大模型必须联网调用或需要高端硬件才能运行的痛点,让日常设备也能成为强大的 AI 助手。无论是希望构建本地知识库的开发者,还是单纯想体验私有化 AI 聊天的普通用户,都能从中受益。 技术上,GPT4All 基于高效的 `llama.cpp` 后端,支持多种主流模型架构(包括最新的 DeepSeek R1 蒸馏模型),并采用 GGUF 格式优化推理速度。它不仅提供界面友好的桌面客户端,支持 Windows、macOS 和 Linux 等多平台一键安装,还为开发者提供了便捷的 Python 库,可轻松集成到 LangChain 等生态中。通过简单的下载和配置,用户即可立即开始探索本地大模型的无限可能。

77.3k|★☆☆☆☆|3天前
语言模型开发框架