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