Skip to content

Latest commit

 

History

2,847 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

简体中文 | English

GitHub Documentation

简介


MindSpeed LLM:基于昇腾生态的大语言模型分布式训练套件,旨在为华为 昇腾芯片 生态合作伙伴提供端到端的大语言模型训练方案,包含分布式预训练、分布式指令微调以及对应的开发工具链,如:数据预处理、权重转换、在线推理、基线评估等。

最新消息


更多消息

注意:【beta】表示特性未经过充分验证,若使用存在问题请至issue反馈。

目录结构


MindSpeed LLM 项目代码按照模块化设计原则进行组织,详细介绍参见 项目导读。

MindSpeed-LLM/
 ├── ci                        # 门禁看护
 ├── configs                   # 配置文件目录
 ├── docker                    # Docker构建配置
 ├── docs                      # 项目文档目录
 ├── examples                  # 模型示例脚本
 ├── mindspeed_llm             # 核心代码目录
 ├── pre-commit                # pre-commit钩子配置
 ├── tests                     # 测试用例目录
 ├── convert_ckpt.py           # 权重转换工具
 ├── convert_ckpt_v2.py        # 权重转换工具 v2
 ├── evaluation.py             # 模型评估工具
 ├── inference.py              # 模型推理工具
 ├── inference_fsdp2.py        # FSDP2推理工具
 ├── posttrain_gpt.py          # 后训练流程
 ├── preprocess_data.py        # 数据预处理工具
 ├── preprocess_prompt.py      # 提示词预处理工具
 ├── pretrain_deepseek4.py     # DeepSeek4预训练流程
 ├── pretrain_gpt.py           # 预训练流程
 ├── pretrain_mamba.py         # 预训练mamba模型流程
 ├── rlhf_gpt.py               # RLHF 训练流程
 ├── train_fsdp2.py            # FSDP2 训练流程
 ├── requirements.txt          # Python依赖文件
 ├── setup.py                  # 安装配置文件
 ├── README.md                 # 项目说明文档

文档导航


文档导航提供了 MindSpeed LLM 的完整文档使用指南,包含以下核心内容:

  • 环境安装指导:MindSpeed LLM 的安装配置说明
  • 快速入门:从环境安装到训练拉起的入门指导
  • 模型清单:基于 PyTorch 框架支持的模型列表
  • 特性清单:性能优化和显存优化的特性说明
  • 训练方案:预训练、微调、推理、评估等完整方案
  • 工具链:权重转换、数据集处理、性能采集分析、确定性计算等工具使用说明

版本说明


查看版本说明,了解MindSpeed LLM的最新发布版本以及历史版本,包含对应版本的软件配套表、版本兼容性说明、以及当前版本的更新说明。

安装


快速上手


指导开发者快速启动大语言模型的预训练和微调任务,具体的操作请参考:

支持模型


MindSpeed LLM目前已内置支持百余个业界常用LLM大模型的预训练与微调,支持模型清单可查看:

训练方案与特性


MindSpeed LLM包含分布式预训练、分布式微调等训练方案,具体介绍请参考训练方案与特性说明。

在线推理


特性 Mcore Released 贡献方
流式推理 ✅ ✅ 【Ascend】
Chat对话 ✅ ✅ 【Ascend】
YaRN上下文扩展 ✅ ❌ 【Ascend】

Note

文档表格中的“Released”代表商用版本已发布,“✅”代表支持,“❌”代表不支持。

开源数据集评测


仓库模型基线见开源数据集评测基线

场景 数据集 Mcore Released 贡献方
评测 MMLU ✅ ❌ 【Ascend】
CEval ✅ ❌ 【Ascend】
BoolQ ✅ ❌ 【Ascend】
BBH ✅ ❌ 【Ascend】
AGIEval ✅ ❌ 【Ascend】
HumanEval ✅ ❌ 【Ascend】
CMMLU ✅ ❌ 【Ascend】
GSM8k ✅ ❌ 【Ascend】
HellaSwag ✅ ❌ 【Ascend】
NeedleBench ✅ ❌ 【Ascend】

开发工具链


权重转换

MindSpeed LLM支持Huggingface、Megatron-core两种格式的权重互转,支持LoRA权重合并。权重转换特性参数和使用说明参考权重转换。

源格式 目标格式 切分特性 LoRA 贡献方 Released
Huggingface Megatron-core tp、pp、dpp、vpp、cp、ep、loop layer ❌ 【Ascend】 ❌
Megatron-core Huggingface ✅
Megatron-core tp、pp、dpp、vpp、cp、ep、loop layer ✅

数据预处理

MindSpeed LLM支持预训练、指令微调等多种任务的数据预处理。

任务场景 数据集 Mcore Released 贡献方
预训练 预训练数据处理 ✅ ✅ 【Ascend】
微调 Alpaca风格 ✅ ✅
ShareGPT风格 ✅ ✅

性能采集

场景 特性 Mcore Released 贡献方
性能采集 基于昇腾芯片采集 profiling 数据 ✅ ❌ 【Ascend】

高可用性

场景 特性 Mcore Released 贡献方
高可用性 基于昇腾芯片开启确定性计算 ✅ ❌ 【Ascend】

版本维护策略


MindSpeed LLM版本有以下五个维护阶段:

状态 时间 说明
计划 1—3 个月 计划特性
开发 3 个月 开发特性
维护 6-12 个月 合入所有已解决的问题并发布版本,针对不同的MindSpeed LLM版本采取不同的维护策略,常规版本和长期支持版本维护周期分别为6个月和12个月
无维护 0—3 个月 合入所有已解决的问题,无专职维护人员,无版本发布
生命周期终止(EOL) N/A 分支不再接受任何修改

MindSpeed LLM已发布版本维护策略:

MindSpeed LLM版本 对应标签 维护策略 当前状态 发布时间 后续状态 EOL日期
26.1.0 v26.1.0 常规版本 开发 预计2026/6/30发布 预计2026/12/30起无维护
26.0.0 v26.0.0 常规版本 维护 2026/3/30 预计2026/9/30起无维护
2.3.0 v2.3.0 常规版本 维护 2025/12/30 预计2026/6/30起无维护
2.2.0 v2.2.0 常规版本 EOL 2025/9/30 生命周期终止 2026/3/30
2.1.0 v2.1.0 常规版本 EOL 2025/6/30 生命周期终止 2025/12/30
2.0.0 v2.0.0 常规版本 EOL 2025/3/30 生命周期终止 2025/9/30
1.0.0 v1.0.0 常规版本 EOL 2024/12/30 生命周期终止 2025/6/30
1.0.RC3 v1.0.RC3.0 常规版本 EOL 2024/09/30 生命周期终止 2025/3/30
1.0.RC2 v1.0.RC2.0 常规版本 EOL 2024/06/30 生命周期终止 2024/12/30
1.0.RC1 v1.0.RC1.0 常规版本 EOL 2024/03/30 生命周期终止 2024/9/30
bk_origin_23 \ Demo EOL 2023 生命周期终止 2024/6/30

未来规划


未来规划会刷新在MindSpeed LLM RoadMap中,欢迎访问LLM最新规划动态。

社区会议


MindSpeed LLM系列TC及SIG会议安排请查看Ascend会议中心

加入我们


为了交流开发经验、分享使用心得、及时获取项目更新,我们创建了MindSpeed LLM社区交流群。无论你是正在使用这个项目,还是有奇思妙想,都欢迎加入。

加入方式:

  1. 直接扫码加入微信交流群(二维码7天有效,定期更新)
  2. 添加昇腾开源小助手,获取群链接,进入MindSpeed LLM社区交流群
MindSpeed LLM社区交流群
MindSpeed LLM 微信群
昇腾开源小助手
昇腾小助手 微信

安全声明


MindSpeed LLM安全声明

免责声明


致MindSpeed LLM使用者

  1. MindSpeed LLM提供的模型仅供您用于非商业目的。
  2. MindSpeed LLM功能依赖的Megatron等第三方开源软件,均由第三方社区提供和维护,因第三方开源软件导致的问题修复依赖相关社区的贡献和反馈。您应理解,MindSpeed LLM仓库不保证对第三方开源软件本身的问题进行修复,也不保证会测试、纠正所有第三方开源软件的漏洞和错误。
  3. 对于各模型,MindSpeed LLM平台仅提示性地向您建议可用于训练的数据集,华为不提供任何数据集,如您使用这些数据集进行训练,请您特别注意应遵守对应数据集的License,如您因使用数据集而产生侵权纠纷,华为不承担任何责任。
  4. 如您在使用MindSpeed LLM模型过程中,发现任何问题(包括但不限于功能问题、合规问题),请在Gitcode提交issue,我们将及时审视并解决。

致数据集所有者

如果您不希望您的数据集在MindSpeed LLM中的模型被提及,或希望更新MindSpeed LLM中的模型关于您的数据集的描述,请在Gitcode提交issue,我们将根据您的issue要求删除或更新您的数据集描述。衷心感谢您对MindSpeed LLM的理解和贡献。

License声明


  • MindSpeed LLM产品的使用许可证,具体请参见LICENSE。
  • MindSpeed LLM工具docs目录下的文档适用CC-BY 4.0许可证,具体请参见LICENSE。

贡献声明


如果您希望向MindSpeed LLM报告问题和贡献代码,具体请参见贡献指南。

FAQ


MindSpeed LLM仓库基本使用过程中常见问题可以参考MindSpeed LLM FAQ。FAQ中未能涵盖的问题,可以在仓库的issues列表中尝试寻找类似问题,或者提交新的issue。

致谢


MindSpeed LLM由华为公司的下列部门以及昇腾生态合作伙伴联合贡献 :

华为公司:

  • 计算产品线:Ascend
  • 公共开发部:NAIE
  • 全球技术服务部:GTS
  • 华为云计算:Cloud

生态合作伙伴:

  • 移动云(China Mobile Cloud):大云震泽智算平台
  • 工商银行软件开发中心大数据人工智能实验室

感谢来自社区的每一个PR,欢迎贡献 MindSpeed LLM。

About

No description, website, or topics provided.

Resources

Contributing

Stars

61 stars

Watchers

3 watching

Forks

Releases

Packages

Used by

Contributors

Languages