drizzlezyk commited on
Commit
e31cd49
·
verified ·
1 Parent(s): e0ae745

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +108 -0
README.md ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 开源盘古 Embedded-1B
2
+ 中文 | [English](README_EN.md)
3
+
4
+ ## 1.简介
5
+
6
+ openPangu-Embedded-1B 是基于昇腾 NPU 从零训练的高效语言模型,参数量为 1B(不含词表Embedding),模型结构采用 26 层 Dense 架构,训练了约 10T tokens。通过昇腾 Atlas 200I A2可用的模型架构设计、数据和训练策略优化,openPangu-Embedded-1B 在保持端侧运行的要求下达到了较高的精度。
7
+
8
+ ## 2. 模型架构
9
+
10
+ openPangu-Embedded-1B 是一个为端侧设备运行而设计的高效快思考语言模型,支持昇腾 Atlas 200I A2。
11
+
12
+
13
+ | | openPangu-Embedded-1B |
14
+ | :---------------------------: | :----------------: |
15
+ | **Architecture** | Dense |
16
+ | **Parameters (Non-Embedding)** | 1B |
17
+ | **Number of Layers** | 26 |
18
+ | **Hidden Dimension** | 1536 |
19
+ | **Attention Mechanism** | GQA |
20
+ | **Number of Attention Heads** | 12 for Q, 6 for KV |
21
+ | **Vocabulary Size** | 153k |
22
+ | **Context Length (Natively)** | 32k |
23
+ | **Training Tokens** | 10T |
24
+
25
+
26
+
27
+ ## 3. 测评结果
28
+
29
+ | 评测集 | 测评指标 | 快思考 |
30
+ |:---: |:---: |:---: |
31
+ | **通用能力** | |
32
+ | MMLU | Acc | 60.72 |
33
+ | CMMLU | Acc | 51.99 |
34
+ | C-Eval | Acc | 60.98 |
35
+ | IF-Eval | Prompt Strict | 56.56 |
36
+ | CLUEWSC | Acc | 68.55 |
37
+ | **数学&推理** | |
38
+ | GSM8K | Acc | 66.72 |
39
+ | MATH-500 | Acc | 52.00 |
40
+ | DROP | F1 | 50.31 |
41
+ | **代码能力** | |
42
+ | MBPP | Pass@1 | 54.09 |
43
+ | HumanEval | Pass@1 | 56.71 |
44
+
45
+ **注:** 评测过程中system prompt 为空。
46
+
47
+
48
+ ## 4. 部署和使用
49
+
50
+ ### 4.1 环境准备
51
+
52
+ ##### 硬件规格
53
+ Atlas 800T A2 (64GB),驱动与固件安装包获取请参照 [[Atlas 800T A2](https://www.hiascend.com/hardware/firmware-drivers/community?product=4&model=26&cann=8.2.RC1.alpha003&driver=Ascend+HDK+25.0.RC1)]。
54
+
55
+ ##### 软件环境
56
+
57
+ - 操作系统:Linux(推荐 openEuler>=24.03)
58
+ - CANN==8.1.RC1,安装准备及流程请参照 [[CANN Install](https://www.hiascend.com/document/detail/zh/CANNCommunityEdition/82RC1alpha002/softwareinst/instg/instg_0001.html?Mode=PmIns&OS=Ubuntu&Software=cannToolKit)]
59
+ - python==3.10
60
+ - torch==2.1.0
61
+ - torch-npu==2.1.0.post12
62
+ - transformers==4.53.2
63
+
64
+ 以上软件配套经过验证,理论可以支持更高版本,如有疑问,可以提交 issue。
65
+
66
+ ### 4.2 权重完整性校验
67
+
68
+ 请参考以下方法对下载内容进行完整性校验,hash 值存储在 checklist.chk 文件中。
69
+
70
+ ```
71
+ #!/usr/bin/env bash
72
+ ARCH=$(uname -m)
73
+ MODEL_PATH="${TARGET_FOLDER}/${MODEL_FOLDER_PATH}"
74
+ cd "$MODEL_PATH" || exit 1
75
+ if [ "$ARCH" = "arm64" ]; then
76
+ sha256sum checklist.chk
77
+ else
78
+ sha256sum -c checklist.chk
79
+ fi
80
+ ```
81
+ ### 4.3 推理样例
82
+ 下述内容提供 openPangu-Embedded-1B 在 `transformers` 框架上进行推理的一个简单示例:
83
+ >运行前请修改 generate.py,添加模型路径。
84
+ ```bash
85
+ cd inference
86
+ python generate.py
87
+ ```
88
+
89
+
90
+ ### 4.4 使用推理框架
91
+ **vllm_ascend:** 参考[[vllm_ascend_for_openpangu_embedded_1b.zh]](inference/vllm_ascend_for_openpangu_embedded_1b.zh.md)
92
+
93
+ **昇腾 Atlas 200I A2推理:** openPangu-Embedded-1B 模型推理已适配昇腾 MindIE 2.2.T10([[下载链接]](https://support.huawei.com/enterprise/zh/ascend-computing/mindie-pid-261803968/software/266130647?idAbsPath=fixnode01|23710424|251366513|254884019|261408772|261803968)),支持 OrangePi AIpro (昇腾 Atlas 200I A2) 推理部署。届时可前往 [[昇腾社区ModelZoo]](https://gitee.com/ascend/ModelZoo-PyTorch/blob/master/MindIE/LLM/Pangu/openPangu-Embedded-1B-OrangePi/README.md) 下载适配,下载镜像前需要申请权限,耐心等待权限申请通过后,根据指南下载对应版本文件和安装指导完成推理部署。
94
+
95
+ ## 5. 模型许可证
96
+
97
+ 除文件中对开源许可证另有约定外,openPangu-Embedded-1B 模型根据 OPENPANGU MODEL LICENSE AGREEMENT VERSION 1.0 授权,旨在允许使用并促进人工智能技术的进一步发展。有关详细信息,请参阅模型存储库根目录中的 [LICENSE](LICENSE) 文件。
98
+
99
+
100
+ ## 6. 免责声明
101
+ 由于 openPangu-Embedded-1B(“模型”)所依赖的技术固有的技术限制,以及人工智能生成的内容是由盘古自动生成的,华为无法对以下事项做出任何保证:
102
+ - 尽管该模型的输出由 AI 算法生成,但不能排除某些信息可能存在缺陷、不合理或引起不适的可能性,生成的内容不代表华为的态度或立场;
103
+ - 无法保证该模型 100% 准确、可靠、功能齐全、及时、安全、无错误、不间断、持续稳定或无任何故障;
104
+ - 该模型的输出内容不构成任何建议或决策,也不保证生成的内容的真实性、完整性、准确性、及时性、合法性、功能性或实用性。生成的内容不能替代医疗、法律等领域的专业人士回答您的问题。生成的内容仅供参考,不代表华为的任何态度、立场或观点。您需要根据实际情况做出独立判断,华为不承担任何责任。
105
+
106
+ ## 7. 反馈
107
+
108
+ 如果有任何意见和建议,请提交issue或联系 openPangu@huawei.com。