华为开源7B参数openPangu-Embedded-7B-v1.1

AI项目和框架5小时前发布 ai-tab
4 0

1. 简介

近日,华为正式在开源社区GitCode上发布了其最新的开源大语言模型——openPangu-Embedded-7B-v1.1。该模型以其70亿(7B)参数的规模和一项突破性的可自由切换思维模式(Thinking Mode)功能,迅速吸引了全球开发者与人工智能研究者的目光,为嵌入式应用和边缘计算场景下的AI能力带来了新的可能。

openPangu-Embedded-7B-V1.1 是基于昇腾 NPU 从零训练的高效大语言模型,参数量为 7B(不含词表Embedding)。openPangu-Embedded-7B-V1.1 训练了约 25T tokens,具备快慢思考融合与自适应切换能力。

华为盘古大模型

2. 模型架构表格

openPangu-Embedded-7B-V1.1
ArchitectureDense
Parameters (Non-Embedding)7B
Number of Layers34
Hidden Dimension12800
Attention MechanismGQA
Number of Attention Heads32 for Q,8 for KV
Vocabulary Size153k
Context Length (Natively)32k
Pretraining Tokens25T

3. 测评结果

测评集测评指标慢思考v1.0慢思考v1.1自适应v1.1
通用能力
MMLU-ProExact Match76.3275.5472.81
CMMLUAcc75.5972.9472.18
ArenaHard_v0.1w/o style control85.8088.0084.60
C-EvalAcc83.0584.9283.33
GPQA-DiamondAvg@470.5473.2373.74
数学能力
MATH-500Avg@195.0097.0096.00
AIME24Avg@1671.5779.3879.02
AIME25Avg@1658.2470.0070.21
代码能力
LiveCodeBenchAvg@2 (08/24~01/25)54.0458.2758.27
MBPP+Avg@276.0676.4675.66

注: 评测过程中system prompt 为空,且不添加任何额外的思维链(CoT)提示。评测采用 128k 的序列长度进行。

除精度外,我们还在部分数据集上统计了模型的输出长度,通过数据质量驱动的学习策略,自适应快慢思考可以在基本不影响精度地前提下,有效地在简单任务上自动切换部分输出为快思考,大幅缩短平均输出思维链长度(Length);在难任务通过保持慢思考能力,精度持平纯慢思考模型。

测评集测评指标慢思考v1.1自适应v1.1
通用能力
CMMLUAcc72.9472.18
Length25741338
C-EvalAcc84.9283.33
Length24841723
数学能力
AIME24Avg@1679.3879.02
Length4822949656
代码能力
LiveCodeBenchAvg@2 (08/24~01/25)58.2758.27
Length5814059307

该模型在多项中英文权威基准评测(如 C-Eval, MMLU, AGIEval 等)中均取得了同等参数量模型中的领先成绩,尤其在知识问答、文本摘要、代码生成等任务上表现优异。其名称中的“Embedded”预示着它对嵌入式设备部署的优化,但也同样适用于服务器端的各种应用。

其主要特点包括:

  • 强大的基础能力:在通用知识和推理任务上表现稳健。
  • 出色的代码能力:能够理解和生成多种编程语言的代码,是开发者的得力助手。
  • 优化的上下文长度:支持更长的文本输入,便于处理长文档和理解复杂语境。
  • 完全开源:模型权重完全开放,允许任何开发者自由使用、研究和商业化,遵循宽松的开源协议。

4. 部署和使用

4.1 环境准备

硬件规格

Atlas 800T A2 (64GB),驱动与固件安装包获取请参照 [Atlas 800T A2]。

软件环境
  • 操作系统:Linux(推荐 openEuler>=24.03)
  • CANN==8.1.RC1,安装准备及流程请参照 [CANN Install]
  • python==3.10
  • torch==2.1.0
  • torch-npu==2.1.0.post12
  • transformers==4.53.2

以上软件配套经过验证,理论可以支持更高版本,如有疑问,可以提交 issue。

4.2 权重完整性校验

请参考以下方法对下载内容进行完整性校验,hash 值存储在 checklist.chk 文件中。

#!/usr/bin/env bash
ARCH=$(uname -m)
MODEL_PATH="${TARGET_FOLDER}/${MODEL_FOLDER_PATH}"
cd "$MODEL_PATH" || exit 1
if [ "$ARCH" = "arm64" ]; then
    sha256sum checklist.chk
else
    sha256sum -c checklist.chk
fi

4.3 推理样例

下述内容提供 openPangu-Embedded-7B-V1.1 在 transformers 框架上进行推理的一个简单示例:

运行前请修改 generate.py,添加模型路径。

cd inference
python generate.py

openPangu-Embedded-7B-V1.1 模型默认为慢思考模式,可以通过以下手段切换至快慢自适应切换/快思考模式:

  • 在代码实例generate.py中,auto_thinking_promptno_thinking_prompt变量的定义展示了切换至快慢自适应或快思考模式的具体实现:通过在用户输入末尾添加/auto_think/no_think标记,可将当前轮次切换至快慢自适应切换/快思考模式。

4.4 使用推理框架

vllm_ascend:参考[vllm_ascend_for_openpangu_embedded_7b.zh]

5. 模型许可证

除文件中对开源许可证另有约定外,openPangu-Embedded-7B-V1.1 模型根据 OPENPANGU MODEL LICENSE AGREEMENT VERSION 1.0 授权,旨在允许使用并促进人工智能技术的进一步发展。有关详细信息,请参阅模型存储库根目录中的 LICENSE文件。

© 版权声明
trae-字节旗下AI代码助手

相关文章

暂无评论

头像
none
暂无评论...