通义千问Qwen2.5-Coder 全系列来咯!强大、多样、实用

简介: 千问团队开源了强大的 Qwen2.5-Coder 系列模型,涵盖 0.5B 到 32B 六种尺寸,旨在推动开放代码模型的发展。该系列模型在代码生成、修复和推理等方面表现出色,支持多种编程语言,并在多个基准测试中达到 SOTA 水平。此外,Qwen2.5-Coder 还提供了丰富的应用场景,如代码助手、Artifacts 和 Interpreter,满足不同开发者的需求。

欢迎扫描文末二维码,关注「阿里云开发者」公众号,了解更多技术干货,关于阿里的技术创新均呈现于此。

1. 引言

千问团队开源「强大」、「多样」、「实用」的 Qwen2.5-Coder 全系列,致力于持续推动 Open Code LLMs 的发展。


● 强大:Qwen2.5-Coder-32B-Instruct 成为目前 SOTA 的开源代码模型,代码能力追平 GPT-4o。在展现出强大且全面的代码能力的同时,具备良好的通用和数学能力;


多样:在之前开源的两个尺寸 1.5B/7B 的基础上,本次开源共带来四个尺寸的模型,包括 0.5B/3B/14B/32B。截止目前 Qwen2.5-Coder 已经覆盖了主流的六个模型尺寸,以满足不同开发者的需要;


● 实用:在两种场景下探索 Qwen2.5-Coder 的实用性,包括代码助手和 Artifacts,一些样例展示出 Qwen2.5-Coder 在实际场景中应用的潜力。


2. 强大:代码能力达到开源模型SOTA

image.png

代码生成:Qwen2.5-Coder-32B-Instruct 作为本次开源的旗舰模型,在多个流行的代码生成基准(EvalPlus, LiveCodeBench, BigCodeBench)上都取得了开源模型中的最佳表现,并且达到和 GPT-4o 有竞争力的表现。


● 代码修复:代码修复是一个重要的编程能力,Qwen2.5-Coder-32B-Instruct 可以帮助用户修复代码中的错误,让编程 更加高效。Aider 是流行的代码修复的基准,Qwen2.5-Coder-32B-Instruct 达到 73.7 分,在 Aider 上的表现与 GPT-4o 相当。


● 代码推理:代码推理指的是模型能否学习代码执行的过程,准确的预测模型的输入与输出。上个月发布的 Qwen2.5-Coder-7B-Instruct 已经在代码推理能力上展现出了不俗的表现,本次 32B 模型在代码推理上更进一步。

image.png

多编程语言:智能的编程助手应该熟悉所有编程语言,Qwen2.5-Coder-32B-Instruct 在 40 多种编程语言上表现出色, 在 McEval 上取得了 65.9 的分数,其中 Haskell, Racket 等语言表现令人印象深刻,这得益于在预训练阶段独特的 数据清洗和配比。

image.png

另外,Qwen2.5-Coder-32B-Instruct 的多编程语言的代码修复能力仍然令人惊喜,这将有助于用户理解和修改自己熟悉的 编程语言,极大缓解陌生语言的学习成本。与 McEval 类似,MdEval 是多编程语言的代码修复基准,Qwen2.5-Coder-32B-Instruct MdEval 上取得了 75.2 的分数,在所有开源模型中排名第一。

image.png

● 人类偏好对齐:为了检验 Qwen2.5-Coder-32B-Instruct 在人类偏好上的对齐表现,构建了一个来自内部标注的代码偏 好评估基准 Code Arena(类似 Arena Hard)。采用 GPT-4o 作为偏好对齐的评测模型,采用 'A vs. B win' 的评测方 式,即在测试集实例中,模型 A 的分数超过模型 B 的百分比。下图结果表现出 Qwen2.5-Coder-32B-Instruct 在偏 好对齐方面的优势。

image.png


3. 全面:丰富的模型尺寸

本次 Qwen2.5-Coder 开源了丰富的模型尺寸,共包含 0.5B/1.5B/3B/7B/14B/32B 六个尺寸,不仅能够满足开发者在不同 资源场景下的需求,还能给研究社区提供良好的实验场。下表是详细的模型信息:

image.png

一直相信 Scaling Law 哲学。评估了不同尺寸的 Qwen2.5-Coder 在所有数据集上的表现,以验证 Scaling 在 Code LLMs 上的有效性。


对于每一个尺寸,都开源了 Base 和 Instruct 模型,其中 Instruct 模型作是一个可以直接聊天的官方对齐模型,Base 模 型作为开发者微调自己模型的基座。


下面是不同尺寸的 Base 模型表现:

image.png

下面是不同尺寸的 Instruct 模型表现:

image.png


为了更加直观,展示了不同尺寸 Qwen2.5-Coder 和其他开源模型在核心数据集上的对比。


● 对于 Base 模型,选择 MBPP-3shot 作为评估指标,大量的实验表明,MBPP-3shot 更适合评估基础模型,且能够和模型的真实效果有较好的相关性。


● 对于 Instruct 模型,选择 LiveCodeBench 最新的 4 个月(2024.07 - 2024.11)的题目作为评估,这些最新公布的,不可能泄露到训练集的题目,能够反映模型的 OOD 能力。


模型尺寸和模型效果之间符合预期的存在正相关,并且 Qwen2.5-Coder 在所有尺寸下都取得了 SOTA 的表现,这鼓励着继续探索更大尺寸的 Coder。

image.png

模型许可


Qwen2.5-Coder的0.5B、1.5B、7B、14B、32B模型均采用Apache 2.0许可证,3B模型使用“Research Only”许可。

4. 模型链接和体验

Qwen2.5-Coder 模型链接


模型集合 demo 链接


小程序体验:

魔搭社区ModelScope,,

通义千问2.5-代码-demo

小程序

Artifacts 体验链接


5. 模型推理

transformers: 单卡运行 Qwen2.5-32B-Instrtuct 量化模型。


from modelscope import AutoModelForCausalLM, AutoTokenizer
model_name = "Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4"
model = AutoModelForCausalLM.from_pretrained( model_name,
torch_dtype="auto",
device_map="auto" )
tokenizer = AutoTokenizer.from_pretrained(model_name)
prompt = "write a quick sort algorithm."
messages = [
{"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
{"role": "user", "content": prompt} ]
text = tokenizer.apply_chat_template( messages,
tokenize=False,
add_generation_prompt=True )
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
generated_ids = model.generate( **model_inputs,
max_new_tokens=512 )
generated_ids = [
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids) ]
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

显存占用:

image.png

Ollama:一行命令使用 Ollama 运行魔搭 Qwen2.5-Coder GGUF 模型

#设置下启用
ollama serve
#ollama run ModelScope 任意 GGUF 模型
ollama run modelscope.cn/Qwen/Qwen2.5-32B-Instruct-GGUF

在安装了 Ollama 的环境上(建议使用>=0.3.12 版本),直接通过上面的命令行,就可以直接在本地运行 模型。


vLLM,推理加速

pip install vllm -U
export VLLM_USE_MODELSCOPE=True vllm serve Qwen/Qwen2.5-Coder-7B-Instruct
# 推理代码:
from openai import OpenAI
# Modify OpenAI's API key and API base to use vLLM's API server. openai_api_key = "EMPTY"
openai_api_base = "http://localhost:8000/v1"
client = OpenAI(
api_key=openai_api_key,
base_url=openai_api_base, )
completion = client.completions.create(model="Qwen/Qwen2.5-Coder-1.5B-Instruct",print("Completion result:", completion)
prompt="San Francisco is a")

6. 模型微调

我们介绍使用 ms-swift 对 qwen2.5-coder 进行自我认知微调,并对微调后模型进行推理。

swift 开源地址

自我认知数据集链接

如果需要使用其他数据集进行微调,只需要修改即可。

自定义 dataset 支持传入本地路径、modelscope 和 huggingface 中 的 dataset_id 。

文 档 可 以 查 看

在开始微调之前,请确保您的环境已正确安装:

# 安装 ms-swift
git clone https://github.com/modelscope/ms-swift.git
cd swift
pip install -e .[llm]
# 微调脚本:
# Experimental environment: A10, 3090, V100, ... # 15GB GPU memory 
CUDA_VISIBLE_DEVICES=0 swift sft \
  --model_type qwen2_5-coder-3b-instruct \ 
  --model_id_or_path qwen/Qwen2.5-Coder-3B-Instruct \ 
  --dataset swift/self-cognition#500 \
              AI-ModelScope/Magpie-Qwen2-Pro-200K-Chinese#500 \
              AI-ModelScope/Magpie-Qwen2-Pro-200K-English#500 \ 
  --logging_steps 5 \
  --max_length 4096 \
  --learning_rate 1e-4 \
  --output_dir output \ 
  --lora_target_modules ALL \ 
  --model_name 小黄 'Xiao Huang' \ 
  --model_author 魔搭 ModelScope \ 
  --system 'You are a helpful assistant.'

微调显存消耗:

image.png

微调后推理脚本如下,这里的 ckpt_dir 需要修改为训练生成的 last checkpoint 文件夹。

# Experimental environment: A10, 3090, V100, ... # 直接推理
CUDA_VISIBLE_DEVICES=0 swift infer \
  --ckpt_dir output/qwen2_5-coder-3b-instruct/vx-xxx/checkpoint-xxx
# 使用 vLLM 进行推理加速 
CUDA_VISIBLE_DEVICES=0 swift infer \
  --ckpt_dir output/qwen2_5-coder-3b-instruct/vx-xxx/checkpoint-xxx \ 
  --infer_backend vllm 
  --max_model_len 8192 
  --merge_lora true

推理结果:

image.png


7. 模型应用:Cursor,Artifacts和interpreter

实用的 Coder 一直是的愿景,为此本次探索了 Qwen2.5-Coder 在代码助手、 Artifacts 、interpreter 场景下的实际表现。

Qwen2.5-Coder 遇到 Cursor:万能代码小助手


只能代码助手目前已经得到广泛的应用,但目前大多依赖闭源模型,希望 Qwen2.5-Coder 的出现能够为开发者提供一个 友好且强大的选择。


配置 Qwen2.5-Coder-32B-Instruct 的 openai 兼容 api(URL 和 API Key)

image.png

体验 Qwen2.5-Coder 强大的生成/编辑/补全能力吧!(Command+K)

image.png

7.1  Qwen2.5-Coder 遇到Artifacts:prompt编程不是梦

Artifacts 是代码生成的重要应用之一,帮助用户创作一些适合可视化的作品,clone魔搭创空间,即可本地搭建一个Artifacts。

git clone https://www.modelscope.cn/studios/Qwen/Qwen2.5-Coder-Artifacts.git cd Qwen2.5-Coder-Artifacts
pip install -r requirements.txt
pip install gradio
python app.py


7.2 Qwen2.5-Coder 遇到Interpreter:AI操作电脑

在MAC安装环境:

pip install open-interpreter

进入Python环境:


from interpreter import interpreter
interpreter.llm.api_base = "YOUR_BASE_URL" 
interpreter.llm.api_key = "YOUR_API_KEY" 
interpreter.llm.model = "openai/Qwen-Coder-32B-Instruct" 
interpreter.chat("Can you set my system to light mode?")




来源  |  阿里云开发者公众号

作者  |  爱开源的




lQDPJxtDT2qZCe3NAQLNAQKwi5rXw9OESjkHEvZYkMDCAA_258_258.jpg

作者介绍
目录