Code Monkey home page Code Monkey logo

yuan-2.0's Introduction

源2.0

Read this in English.

📔 更为详细的使用信息,可以参考:源2.0 论文

目录

介绍

源2.0 是浪潮信息发布的新一代基础语言大模型。我们开源了全部的3个模型源2.0-102B,源2.0-51B和源2.0-2B。并且我们提供了预训练,微调,推理服务的相关脚本,以供研发人员做进一步的开发。源2.0是在源1.0的基础上,利用更多样的高质量预训练数据和指令微调数据集,令模型在语义、数学、推理、代码、知识等不同具备更强的理解能力。


对本仓库源码的使用遵循开源许可协议 Apache 2.0

源2.0模型支持商用,不需要申请授权,请您了解并遵循《源2.0模型许可协议》,勿将开源模型和代码及基于开源项目产生的衍生物用于任何可能给国家和社会带来危害的用途以及用于任何未经过安全评估和备案的服务。

尽管模型在训练时我们已采取措施尽力确保数据的合规性和准确性,但模型参数量巨大且受概率随机性因素影响,我们无法保证输出内容的准确性,且模型易被输入指令所误导,本项目不承担开源模型和代码导致的数据安全、舆情风险或发生任何模型被误导、滥用、传播、不当利用而产生的风险和责任。您将对通过使用、复制、分发和修改模型等方式利用该开源项目所产生的风险与后果,独自承担全部责任。

快速启动

环境配置

我们建议使用有我们提供的最新的docker镜像文件.

我们可以通过下面命令启动容器:

docker load < ./yuan_v2.0.tar
docker run --gpus all -it -v /path/to/yuan_2.0:/workspace/yuan_2.0 -v /path/to/dataset:/workspace/dataset -v /path/to/checkpoints:/workspace/checkpoints yuan_v2.0:latest

数据预处理

我们提供了数据预处理的脚本,参考数据预处理说明文档.

预训练

我们提供了用于预训练的文档和 example的脚本,具体使用方法可以参考预训练说明文档.

模型微调

请参考指令微调 源2.0 指令微调示例

请注意,不同的微调脚本对应的模型并不相同,请根据需要选择对应的模型。

模型

我们提供了源2.0有监督微调的模型文件。模型文件可以通过下面的链接下载得到:

Model 序列长度 Download Link
源2.0-102B 4K here
源2.0-51B 4K here
源2.0-2B 8K here

源2.0-2B模型支持的序列长度为8192个tokens,源2.0-51B和源2.0-102B模型支持的序列长度为4096个tokens,可以根据用户设备的内存大小设置 --max-position-embeddings--seq-length 的值。

评测结果

我们提供了HumanEvalAGIEval-MathGSM-8KTruthfulQA的评估脚本。在4个典型任务上,我们用Yuan-2.0不同版本模型上进行了性能测试。

Model GSM8K AGIEval-GK-Math-QA AGIEval-GK-Math-Cloze HumanEval TurthfulQA
GPT-4 92% 47.0% 16.1% 86.6% 59%
Chat-GPT 68.6%* 36.5% 7.3% 66.5%* 34%*
Llama2 56.8% - - 29.9% -
源2.0-102B 76.6% 38.7% 13.5% 67.1% 58%
源2.0-102B-SC 86.2% 45.5% 15.2% 77.4% -

* 使用与源2.0完全相同的输入数据对ChatGPT进行测试,时间2023年11月

代码调用

考虑到推理服务的效率,源2.0-51B和源2.0-102B模型在启动推理服务之前,需要将模型转换成只有张量并行的模型文件。可以参考文档

可以通过调用推理服务,向推理服务发送请求实现模型的调用,源2.0 推理服务

yuan-2.0's People

Contributors

zhaoxudong01 avatar shawn-ieitsystems avatar zhaoxudong01-ieisystem avatar

Stargazers

Weixiao Zhou avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.