CodeGeeX

CodeGeeX

强大的多语言代码生成与翻译模型

CodeGeeX是一个拥有130亿参数的大规模多语言代码生成模型。该模型可生成Python、C++、Java等主流编程语言的可执行代码,并支持跨语言代码翻译。CodeGeeX提供免费的VS Code和Jetbrains IDE扩展,在HumanEval-X多语言基准测试中表现优异。项目完全开源,同时支持Ascend和NVIDIA平台,为开发者提供高效的代码生成工具。

CodeGeeX代码生成多语言人工智能编程助手Github开源项目
<img src="https://yellow-cdn.veclightyear.com/2b54e442/0b2a8c29-e48b-448f-b927-b37825e107ba.png"> <p align="center"> 🏠 <a href="https://codegeex.cn" target="_blank">主页</a> | 📖 <a href="https://models.aminer.cn/codegeex/blog/" target="_blank">博客</a> | 🪧 <a href="https://models.aminer.cn/codegeex/playground" target="_blank">演示</a> | 🤖 <a href="https://codegeex.cn/download/request" target="_blank">下载模型</a> | 📄 <a href="https://arxiv.org/abs/2303.17568" target="_blank">论文</a> | 🌐 <a href="README_zh.md" target="_blank">中文</a> </p> <p align="center"> 🛠 支持 <a href="https://marketplace.visualstudio.com/items?itemName=aminer.codegeex" target="_blank">VS Code</a>、<a href="https://plugins.jetbrains.com/plugin/20587-codegeex" target="_blank">Jetbrains</a>、<a href="https://plugins.jetbrains.com/plugin/20587-codegeex" target="_blank">Cloud Studio</a> | 👋 加入我们的 <a href="https://discord.gg/8gjHdkmAN6" target="_blank">Discord</a>、<a href="https://join.slack.com/t/codegeexworkspace/shared_invite/zt-1s118ffrp-mpKKhQD0tKBmzNZVCyEZLw" target="_blank">Slack</a>、<a href="https://t.me/+IipIayJ32B1jOTg1" target="_blank">Telegram</a>、<a href="resources/zh/wechat.md"target="_blank">微信</a> </p>

🌟 最新一代 CodeGeeX4 模型已经正式开源。

CodeGeeX:多语言代码生成模型

我们介绍CodeGeeX,这是一个拥有130亿参数的大规模多语言代码生成模型,在包含20多种编程语言的大型代码语料库上进行了预训练。截至2022年6月22日,CodeGeeX已经在由1,536个昇腾910 AI处理器组成的集群上训练了超过8500亿个标记。CodeGeeX具有以下几个独特特征:

  • 多语言代码生成:CodeGeeX在生成几种主流编程语言的可执行程序方面表现出色,包括Python、C++、Java、JavaScript、Go等。演示
  • 跨语言代码翻译:CodeGeeX支持不同语言之间的代码片段翻译。只需一次点击,CodeGeeX就可以以高精度将程序转换为任何预期的语言。演示
  • 可定制的编程助手:CodeGeeX在VS Code扩展市场上免费提供。它支持代码补全、解释、总结等功能,为用户提供更好的编码体验。VS Code扩展
  • 开源和跨平台:所有代码和模型权重都公开可用于研究目的。CodeGeeX支持昇腾和NVIDIA平台。它支持在单个昇腾910、NVIDIA V100或A100上进行推理。申请模型权重

HumanEval-X用于真实多语言基准测试。 为了帮助标准化多语言代码生成和翻译的评估,我们开发并发布了HumanEval-X基准。HumanEval-X是一个新的多语言基准,包含820个人工编写的编码问题,涉及5种编程语言(Python、C++、Java、JavaScript和Go),每个问题都有相关的测试和解决方案。使用方法 🤗 在HuggingFace上可用

<img src="https://yellow-cdn.veclightyear.com/2b54e442/bcecf6d0-95e9-425b-94e8-86ed7c0c50ab.png"> <p align="center"><i>与其他开源多语言基线相比,CodeGeeX在平均性能上达到了最高水平。</i> </p>

新闻

  • 🌟 2023-07-24CodeGeeX2 已经发布,更强大、更快速、更轻量。支持100+种语言和许多新功能。

  • 2023-5-16:CodeGeeX论文已被KDD 2023,长滩接受,将在会议期间进行展示。

  • 2023-03-30:CodeGeeX论文现已在arxiv上发布。

  • 2023-02-14:CodeGeeX现在支持Cloud Studio,这是腾讯的一款出色的Web IDE。点击此页面顶部的徽章,快速启动环境来测试CodeGeeX。

  • 2023-02-13:非常感谢OneFlow团队为CodeGeeX的推理添加oneflow后端(在FP16下甚至比FasterTransformer更快!)。查看更多详情这里

  • 2023-02:我们正在举办CodeGeeX "用AI编程" 黑客马拉松,基于CodeGeeX设计酷炫的应用并赢取奖品(RTX 4090、大疆无人机等)!

  • 2022-12-31:我们在codegeex-fastertransformer中发布了CodeGeeX的FasterTransformer版本。INT8加速版本达到了平均<15ms/token的速度。祝大家新年快乐!

  • 2022-12-13:我们在codegeex-vscode-extension中发布了CodeGeeX VS Code扩展的源代码。按照快速入门开始开发。

  • 2022-12-11:CodeGeeX现在可用于Jetbrains IDEs(IntelliJ IDEA、PyCharm、GoLand、CLion等),在这里下载。

  • 2022-12-04:我们发布了量化(需要更少的GPU RAM:27GB -> 15GB)和模型并行(可以在多个RAM<8G的GPU上运行)的源代码。

  • 2022-09-30:我们发布了适用于昇腾和NVIDIA平台的跨平台源代码和模型权重。

入门指南

CodeGeeX最初是在Mindspore中实现并在昇腾910 AI处理器上训练的。我们提供了一个基于Megatron-LM的与torch兼容的版本,以便于在GPU平台上使用。

安装

需要Python 3.7+ / CUDA 11+ / PyTorch 1.10+ / DeepSpeed 0.6+。通过以下方式安装codegeex包:

git clone git@github.com:THUDM/CodeGeeX.git cd CodeGeeX pip install -e .

或者使用CodeGeeX docker快速设置环境(需要安装nvidia-docker):

docker pull codegeex/codegeex:latest # 要启用GPU支持,使用--device指定设备ID docker run --gpus '"device=0,1"' -it --ipc=host --name=codegeex codegeex/codegeex

模型权重

通过此链接申请并下载模型权重。您将通过邮件收到urls.txt,其中包含临时下载链接。我们建议您使用aria2通过以下命令下载(请确保您有足够的磁盘空间来下载检查点(约26GB)):

aria2c -x 16 -s 16 -j 4 --continue=true -i urls.txt

运行以下命令获取完整的模型权重:

cat codegeex_13b.tar.gz.* > codegeex_13b.tar.gz tar xvf codegeex_13b.tar.gz

在GPU上进行推理

尝试用CodeGeeX生成第一个程序。首先,在configs/codegeex_13b.sh中指定模型权重的路径。其次,将提示(自然语言描述或代码片段)写入文件,例如tests/test_prompt.txt,然后运行以下脚本:

# 在单个GPU上(RAM大于27GB) bash ./scripts/test_inference.sh <GPU_ID> ./tests/test_prompt.txt # 使用量化(RAM大于15GB) bash ./scripts/test_inference_quantized.sh <GPU_ID> ./tests/test_prompt.txt # 在多个GPU上(RAM大于6GB,需要首先将ckpt转换为MP_SIZE分区) bash ./scripts/convert_ckpt_parallel.sh <LOAD_CKPT_PATH> <SAVE_CKPT_PATH> <MP_SIZE> bash ./scripts/test_inference_parallel.sh <MP_SIZE> ./tests/test_prompt.txt

VS Code和Jetbrains扩展指南

基于CodeGeeX,我们还开发了VS Code和Jetbrains IDEs的免费扩展,未来还会有更多。

对于VS Code,在市场中搜索"codegeex"或在这里安装。详细说明可以在VS Code扩展指南中找到。对于开发者,我们还在codegeex-vscode-extension中发布了源代码,请按照快速入门开始开发。

对于Jetbrains IDEs,在插件中搜索"codegeex"或在这里安装。 确保您的IDE版本是2021.1或更高。CodeGeeX现在支持IntelliJ IDEA、PyCharm、GoLand、CLion、Android Studio、AppCode、Aqua、DataSpell、DataGrip、Rider、RubyMine和WebStorm。

CodeGeeX:架构、代码语料库和实现

架构:CodeGeeX是一个基于transformer的大规模预训练编程语言模型。它是一个从左到右的自回归解码器,接受代码和自然语言作为输入,并预测下一个标记的概率。CodeGeeX包含40个transformer层,自注意力块的隐藏大小为5,120,前馈层的隐藏大小为20,480,使其规模达到130亿参数。它支持最大序列长度为2,048。

<img src="https://yellow-cdn.veclightyear.com/2b54e442/48b07af5-476b-4e31-bfce-775308329333.png"> <p align="center"><i><b>左图:</b>CodeGeeX训练数据中编程语言的比例。 <b>右图:</b>CodeGeeX训练步骤与训练损失的关系图。</i></p>

代码语料库:我们的训练数据包含两部分。第一部分来自开源代码数据集,The Pile和[CodeParrot](https://github.com/huggingface/transformers

<p align="center"><i><b>HumanEval-X</b>支持的任务示意图。声明、文档字符串和解决方案分别用红色、绿色和蓝色标记。<b>代码生成</b>使用声明和文档字符串作为输入来生成解决方案。<b>代码翻译</b>使用两种语言的声明,将源语言的解决方案翻译成目标语言。</i></p>

在HumanEval-X中,每种语言的每个样本都包含声明、文档字符串和解决方案,可以通过各种方式组合来支持不同的下游任务,包括生成、翻译、总结等。我们目前专注于两个任务:代码生成代码翻译。对于代码生成,模型使用声明和文档字符串作为输入来生成解决方案。对于代码翻译,模型使用两种语言的声明和源语言的解决方案作为输入,生成目标语言的解决方案。我们在代码翻译过程中移除了描述,以防止模型直接解决问题。对于这两个任务,我们使用Codex中提出的无偏pass@k指标:$\text{pass}@k:= \mathbb{E}[1-\frac{\tbinom{n-c}{k}}{\tbinom{n}{k}}]$,其中$n=200$且$k\in(1,10,100)$。

多语言代码生成

<img src="https://yellow-cdn.veclightyear.com/2b54e442/6c4e39e8-10aa-4563-9701-ce666250857a.png"> <p align="center"><i><b>左图</b>:HumanEval-X中五种语言的代码生成任务的详细pass@k (k=1,10,100)性能。<b>右图</b>:每个模型在所有语言上的平均性能。与InCoder-6.7B、CodeGen-Multi-6B和CodeGen-Multi-16B相比,CodeGeeX实现了最高的平均性能。</i></p>

我们将CodeGeeX与另外两个开源代码生成模型进行了比较,InCoder(来自Meta)和CodeGen(来自Salesforce)。具体来说,考虑了InCoder-6.7B、CodeGen-Multi-6B和CodeGen-Multi-16B。CodeGeeX显著优于规模较小的模型(提高7.5%~16.3%),并且与规模更大的CodeGen-Multi-16B相当(平均性能54.76%对54.39%)。CodeGeeX在各种语言中实现了最佳的平均性能。

跨语言代码翻译

<img src="https://yellow-cdn.veclightyear.com/2b54e442/979cfd28-90d2-425f-9dd0-bcec17b7ea63.png"> <p align="center"><i>HumanEval-X<b>代码翻译</b>任务的结果。每种语言的最佳性能以<b>粗体</b>显示。</i></p>

我们还评估了不同编程语言之间的翻译性能。我们测试了CodeGeeX的零样本性能,以及经过微调的CodeGeeX-13B-FT(使用XLCoST中代码翻译任务的训练集进行微调;原始集合中缺少Go,因此我们添加了一个小集合)。结果表明,模型对语言有偏好,例如,CodeGeeX擅长将其他语言翻译成Python和C++,而CodeGen-Multi-16B更擅长翻译成JavaScript和Go;这可能是由于训练语料库中语言分布的差异。在20对翻译中,我们还观察到A到B和B到A的性能总是负相关,这可能表明当前的模型仍然无法很好地学习所有语言。

如何使用HumanEval-X并为其做出贡献?

有关如何使用HumanEval-X的更多详细信息,请参阅使用说明。我们非常欢迎社区通过添加更多问题或扩展到其他语言来为HumanEval-X做出贡献,请查看HumanEval-X的标准格式并添加拉取请求。

如果您有任何意见或建议,请通过codegeex@aminer.cn告诉我们。

<details> <summary><b>生成示例</b></summary> <img src="https://yellow-cdn.veclightyear.com/2b54e442/e5bab0d1-835e-4046-aab5-b9ffaeeaca06.png"> </details> <details> <summary><b>致谢</b></summary> <br/> 本项目得到国家杰出青年科学基金(No. 61825602)的支持。

主要贡献者

郑钦凯(清华KEG)、夏晓(清华KEG)、邹旭(清华KEG)

贡献者

清华KEG---清华知识工程组:曾奥涵、郑文迪、薛立龙

清华IIIS杨植麟组:刘一峰、陈燕如、徐一宸(北邮,访问清华期间完成工作)

鹏城实验室:陈庆宇、李中奇、范高俊

智谱.AI:薛宇飞、王珊、单杰才、江浩翰、刘路、薛轩、张鹏

昇腾和MindSpore团队:姚逸凡、苏腾、邓启晖、周斌

数据标注

程瑞洁(清华)、余佩楠(清华)、张静瑶(智谱.AI)、黄博文(智谱.AI)、王少宇(智谱.AI)

顾问

杨植麟(清华IIIS)、董宇晓(清华KEG)、陈文光(清华PACMAN)、唐杰(清华KEG)

计算赞助商

鹏城实验室

智谱.AI---一家旨在教机器像人类一样思考的AI创业公司

项目负责人

唐杰(清华KEG & BAAI)

</details>

许可证

我们的代码采用Apache-2.0许可证授权。 我们的模型采用许可证授权。

引用

如果您发现我们的工作有用,请引用:

@inproceedings{zheng2023codegeex,
  title={CodeGeeX: A Pre-Trained Model for Code Generation with Multilingual Benchmarking on HumanEval-X},
  author={Qinkai Zheng and Xiao Xia and Xu Zou and Yuxiao Dong and Shan Wang and Yufei Xue and Zihan Wang and Lei Shen and Andi Wang and Yang Li and Teng Su and Zhilin Yang and Jie Tang},
  booktitle={Proceedings of the 29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining},
  pages={5673--5684},
  year={2023}
}

编辑推荐精选

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
讯飞星火

讯飞星火

深度推理能力全新升级,全面对标OpenAI o1

科大讯飞的星火大模型,支持语言理解、知识问答和文本创作等多功能,适用于多种文件和业务场景,提升办公和日常生活的效率。讯飞星火是一个提供丰富智能服务的平台,涵盖科技资讯、图像创作、写作辅助、编程解答、科研文献解读等功能,能为不同需求的用户提供便捷高效的帮助,助力用户轻松获取信息、解决问题,满足多样化使用场景。

热门AI开发模型训练AI工具讯飞星火大模型智能问答内容创作多语种支持智慧生活
Spark-TTS

Spark-TTS

一种基于大语言模型的高效单流解耦语音令牌文本到语音合成模型

Spark-TTS 是一个基于 PyTorch 的开源文本到语音合成项目,由多个知名机构联合参与。该项目提供了高效的 LLM(大语言模型)驱动的语音合成方案,支持语音克隆和语音创建功能,可通过命令行界面(CLI)和 Web UI 两种方式使用。用户可以根据需求调整语音的性别、音高、速度等参数,生成高质量的语音。该项目适用于多种场景,如有声读物制作、智能语音助手开发等。

Trae

Trae

字节跳动发布的AI编程神器IDE

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

AI工具TraeAI IDE协作生产力转型热门
咔片PPT

咔片PPT

AI助力,做PPT更简单!

咔片是一款轻量化在线演示设计工具,借助 AI 技术,实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT,提供海量模板、智能美化、素材替换等功能,适用于销售、教师、学生等各类人群,能高效制作出高品质 PPT,满足不同场景演示需求。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
材料星

材料星

专业的AI公文写作平台,公文写作神器

AI 材料星,专业的 AI 公文写作辅助平台,为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能,支持 30 + 文稿类型生成,助力快速完成领导讲话、工作总结、述职报告等材料,提升办公效率,是体制打工人的得力写作神器。

openai-agents-python

openai-agents-python

OpenAI Agents SDK,助力开发者便捷使用 OpenAI 相关功能。

openai-agents-python 是 OpenAI 推出的一款强大 Python SDK,它为开发者提供了与 OpenAI 模型交互的高效工具,支持工具调用、结果处理、追踪等功能,涵盖多种应用场景,如研究助手、财务研究等,能显著提升开发效率,让开发者更轻松地利用 OpenAI 的技术优势。

Hunyuan3D-2

Hunyuan3D-2

高分辨率纹理 3D 资产生成

Hunyuan3D-2 是腾讯开发的用于 3D 资产生成的强大工具,支持从文本描述、单张图片或多视角图片生成 3D 模型,具备快速形状生成能力,可生成带纹理的高质量 3D 模型,适用于多个领域,为 3D 创作提供了高效解决方案。

3FS

3FS

一个具备存储、管理和客户端操作等多种功能的分布式文件系统相关项目。

3FS 是一个功能强大的分布式文件系统项目,涵盖了存储引擎、元数据管理、客户端工具等多个模块。它支持多种文件操作,如创建文件和目录、设置布局等,同时具备高效的事件循环、节点选择和协程池管理等特性。适用于需要大规模数据存储和管理的场景,能够提高系统的性能和可靠性,是分布式存储领域的优质解决方案。

下拉加载更多