Skip to content

Commit

Permalink
Merge pull request datawhalechina#257 from Tsumugii24/master
Browse files Browse the repository at this point in the history
Qwen2.5 o1-like reasoning
  • Loading branch information
KMnO4-zx authored Sep 20, 2024
2 parents 6dd8a6a + 605daa8 commit f72927a
Show file tree
Hide file tree
Showing 8 changed files with 305 additions and 12 deletions.
21 changes: 9 additions & 12 deletions models/Qwen2.5/03-Qwen2.5-7B-Instruct vLLM 部署调用.md
Original file line number Diff line number Diff line change
Expand Up @@ -162,10 +162,8 @@ Prompt: '可以给我将一个有趣的童话故事吗?', Generated text: '
- `--served-model-name` 指定服务模型的名称。
- `--max-model-len` 指定模型的最大长度。

这里指定 `--max-model-len=2048` 是因为 `Qwen2.5-7B-Instruct` 模型的最大长度为 `128K`,防止 `vLLM` 初始化 `KV` 缓存时消耗资源过大。

```bash
python -m vllm.entrypoints.openai.api_server --model /root/autodl-tmp/qwen/Qwen2.5-7B-Instruct --served-model-name Qwen2-7B-Instruct --max-model-len=2048
python -m vllm.entrypoints.openai.api_server --model /root/autodl-tmp/qwen/Qwen2.5-7B-Instruct --served-model-name Qwen2.5-7B-Instruct --max-model-len=2048
```

加载完毕后出现如下信息说明服务成功启动
Expand All @@ -185,11 +183,11 @@ curl http://localhost:8000/v1/models
"object": "list",
"data": [
{
"id": "Qwen2-7B-Instruct",
"id": "Qwen2.5-7B-Instruct",
"object": "model",
"created": 1726728585,
"owned_by": "vllm",
"root": "Qwen2-7B-Instruct",
"root": "Qwen2.5-7B-Instruct",
"parent": null,
"max_model_len": 2048,
"permission": [
Expand All @@ -214,14 +212,13 @@ curl http://localhost:8000/v1/models
```

- 使用 `curl` 命令测试 `OpenAI Completions API`
- **注意:**这边有个很关键的点是虽然我们下载了 `Qwen2.5-7B-Instruct` ,但是从上面可以看到 `id` 依然是 `Qwen2-7B-Instruct`


```bash
curl http://localhost:8000/v1/completions \
-H "Content-Type: application/json" \
-d '{
"model": "Qwen2-7B-Instruct",
"model": "Qwen2.5-7B-Instruct",
"prompt": "你好",
"max_tokens": 500,
"temperature": 0
Expand All @@ -235,7 +232,7 @@ curl http://localhost:8000/v1/completions \
"id": "cmpl-e98b85ad1b8942f6959993d644634b0a",
"object": "text_completion",
"created": 1726729800,
"model": "Qwen2-7B-Instruct",
"model": "Qwen2.5-7B-Instruct",
"choices": [
{
"index": 0,
Expand Down Expand Up @@ -292,7 +289,7 @@ ChatCompletionMessage(content='你好!有什么我可以帮助你的吗?', r
curl http://localhost:8000/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "Qwen2-7B-Instruct",
"model": "Qwen2.5-7B-Instruct",
"messages": [
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "你好"}
Expand All @@ -307,7 +304,7 @@ curl http://localhost:8000/v1/chat/completions \
"id": "chat-78357219240d49248afa0f655f85d0fc",
"object": "chat.completion",
"created": 1726730139,
"model": "Qwen2-7B-Instruct",
"model": "Qwen2.5-7B-Instruct",
"choices": [
{
"index": 0,
Expand Down Expand Up @@ -345,7 +342,7 @@ client = OpenAI(
)

chat_outputs = client.chat.completions.create(
model="Qwen2-7B-Instruct",
model="Qwen2.5-7B-Instruct",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Qwen2.5和Qwen2相比有进步和区别"},
Expand All @@ -361,7 +358,7 @@ python vllm_openai_chat_completions.py
得到的返回值如下所示

```
ChatCompletion(id='chat-0c17fb285541409ab6500a41b9312756', choices=[Choice(finish_reason='stop', index=0, logprobs=None, message=ChatCompletionMessage(content='Qwen2.5是基于Qwen2进行的迭代升级,相较于Qwen2,Qwen2.5在多个方面都有所提升和改进。以下是一些主要的区别和进步:\n\n1. **性能优化**:Qwen2.5在处理速度和响应时间上有所提升,能够更快速地理解和生成回答。\n\n2. **语言理解能力增强**:Qwen2.5对自然语言的理解能力进一步提升,能够更好地理解复杂的语言结构和语义,提供更准确的回答。\n\n3. **知识库更新**:Qwen2.5的知识库进行了更新和扩展,涵盖了更多的信息和最新的数据,使得回答更加准确和全面。\n\n4. **多模态处理能力增强**:Qwen2.5在处理图片、音频等多媒体信息方面的能力有所增强,能够更好地理解和生成基于多种输入形式的回答。\n\n5. **个性化和定制化**:Qwen2.5在个性化服务方面有所改进,能够更好地根据用户的需求提供定制化的回答和建议。\n\n6. **安全性增强**:Qwen2.5在安全性方面进行了加强,能够更好地保护用户的隐私和数据安全。\n\n7. **用户体验优化**:Qwen2.5在用户交互设计方面进行了优化,使得用户与AI的交流更加流畅和自然。\n\n这些改进和提升使得Qwen2.5在处理各种复杂任务和满足用户需求方面表现得更加出色,提供更高质量的服务。', refusal=None, role='assistant', function_call=None, tool_calls=[]), stop_reason=None)], created=1726730431, model='Qwen2-7B-Instruct', object='chat.completion', service_tier=None, system_fingerprint=None, usage=CompletionUsage(completion_tokens=304, prompt_tokens=33, total_tokens=337, completion_tokens_details=None), prompt_logprobs=None)
ChatCompletion(id='chat-0c17fb285541409ab6500a41b9312756', choices=[Choice(finish_reason='stop', index=0, logprobs=None, message=ChatCompletionMessage(content='Qwen2.5是基于Qwen2进行的迭代升级,相较于Qwen2,Qwen2.5在多个方面都有所提升和改进。以下是一些主要的区别和进步:\n\n1. **性能优化**:Qwen2.5在处理速度和响应时间上有所提升,能够更快速地理解和生成回答。\n\n2. **语言理解能力增强**:Qwen2.5对自然语言的理解能力进一步提升,能够更好地理解复杂的语言结构和语义,提供更准确的回答。\n\n3. **知识库更新**:Qwen2.5的知识库进行了更新和扩展,涵盖了更多的信息和最新的数据,使得回答更加准确和全面。\n\n4. **多模态处理能力增强**:Qwen2.5在处理图片、音频等多媒体信息方面的能力有所增强,能够更好地理解和生成基于多种输入形式的回答。\n\n5. **个性化和定制化**:Qwen2.5在个性化服务方面有所改进,能够更好地根据用户的需求提供定制化的回答和建议。\n\n6. **安全性增强**:Qwen2.5在安全性方面进行了加强,能够更好地保护用户的隐私和数据安全。\n\n7. **用户体验优化**:Qwen2.5在用户交互设计方面进行了优化,使得用户与AI的交流更加流畅和自然。\n\n这些改进和提升使得Qwen2.5在处理各种复杂任务和满足用户需求方面表现得更加出色,提供更高质量的服务。', refusal=None, role='assistant', function_call=None, tool_calls=[]), stop_reason=None)], created=1726730431, model='Qwen2.5-7B-Instruct', object='chat.completion', service_tier=None, system_fingerprint=None, usage=CompletionUsage(completion_tokens=304, prompt_tokens=33, total_tokens=337, completion_tokens_details=None), prompt_logprobs=None)
```

另外,在以上所有的在请求处理过程中, `API` 后端都会打印相对应的日志和统计信息😊
Expand Down
296 changes: 296 additions & 0 deletions models/Qwen2.5/06-Qwen2.5-7B-Instruct o1-like 推理链实现.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,296 @@
# 06-Qwen2.5-7B-Instruct o1-like 推理链实现

## **OpenAI o1 model 简介**

**OpenAI o1** 系列模型是使用强化学习训练的大语言模型,用于执行复杂推理。o1 模型在回答之前会思考,在向用户做出回应之前可以产生一个长的内部思维链。 o1 模型在科学推理方面表现出色,在编程竞赛(Codeforces)中排名 89%,在美国数学奥林匹克竞赛(AIME)的预选中位列前 500 名,在物理、生物学和化学问题基准测试(GPQA)中超越了人类博士水平。

![06-1](./images/06-1.png)

**OpenAI o1** 系列模型不仅提高了模型的实用性,还为未来AI技术的发展开辟了新的道路。目前,o1模型包括 `o1-preview`` o1-mini` 两个版本,其中 `o1-preview` 适用于解决各个领域的复杂问题,而 `o1-mini` 则速度更快,性价比更高,且更擅长代码领域。

![06-2](./images/06-2.png)

参考文档:

[Reasoning - OpenAI API](https://platform.openai.com/docs/guides/reasoning/quickstart)

[o1 System Card | OpenAI](https://openai.com/index/openai-o1-system-card/)



## 环境准备

本文基础环境如下:

```
----------------
ubuntu 22.04
python 3.12
cuda 12.1
pytorch 2.3.0
----------------
```

> 本文默认学习者已配置好以上 `Pytorch (cuda)` 环境,如未配置请先自行安装。
首先 `pip` 换源加速下载并安装依赖包

```bash
python -m pip install --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

pip install modelscope==1.18.0
pip install openai==1.46.0
pip install tqdm==4.66.2
pip install transformers==4.44.2
pip install vllm==0.6.1.post2
pip install streamlit==1.38.0
```

> 考虑到部分同学配置环境可能会遇到一些问题,我们在AutoDL平台准备了 `Qwen2.5` 的环境镜像,点击下方链接并直接创建 `AutoDL` 示例即可。
> ***https://www.codewithgpu.com/i/datawhalechina/self-llm/Qwen2.5-self-llm***


## 模型下载

使用 `modelscope` 中的 `snapshot_download` 函数下载模型,第一个参数为模型名称,参数 `cache_dir`为模型的下载路径。

先切换到 `autodl-tmp` 目录,`cd /root/autodl-tmp`

然后新建名为 `model_download.py``python` 脚本,并在其中输入以下内容并保存

```python
# model_download.py
from modelscope import snapshot_download
model_dir = snapshot_download('qwen/Qwen2.5-7B-Instruct', cache_dir='/root/autodl-tmp', revision='master')
```

然后在终端中输入 `python model_download.py` 执行下载,这里需要耐心等待一段时间直到模型下载完成。

> 注意:记得修改 `cache_dir` 为你的模型下载路径哦~
![03-1](./images/03-1.png)



## **代码准备**

#### **核心代码**

`/root/autodl-tmp` 路径下新建 `app_qwen.py` 文件并在其中输入以下内容,粘贴代码后请及时保存文件。

```python
# app_qwen.py
import os
import json
import time
import streamlit as st
from openai import OpenAI

client = OpenAI(
api_key="sk-xxx",
base_url="http://localhost:8000/v1",
)

def make_api_call(messages, max_tokens, is_final_answer=False):
for attempt in range(3):
try:
response = client.chat.completions.create(
model="Qwen2.5-7B-Instruct",
messages=messages,
max_tokens=max_tokens,
temperature=0.2,
response_format={"type": "json_object"}
)
content = response.choices[0].message.content
print(f"Raw API response: {content}") # 添加此行来打印原始响应
try:
return json.loads(content)
except json.JSONDecodeError as json_error:
print(f"JSON解析错误: {json_error}")
# 如果JSON解析失败,返回一个包含原始内容的字典
return {
"title": "API Response",
"content": content,
"next_action": "final_answer" if is_final_answer else "continue"
}
except Exception as e:
if attempt == 2:
return {
"title": "Error",
"content": f"Failed after 3 attempts. Error: {str(e)}",
"next_action": "final_answer"
}
time.sleep(1) # 重试前等待1秒

def generate_response(prompt):
messages = [
{"role": "system", "content": """
你是一位具有高级推理能力的专家。你的任务是提供详细的、逐步的思维过程解释。对于每一步:
1. 提供一个清晰、简洁的标题,描述当前的推理阶段。
2. 在内容部分详细阐述你的思维过程。
3. 决定是继续推理还是提供最终答案。
输出格式说明:
输出请严格遵循JSON格式, 包含以下键: 'title', 'content', 'next_action'(值只能为'continue' 或 'final_answer'二者之一)
关键指示:
- 至少使用5个不同的推理步骤。
- 承认你作为AI的局限性,明确说明你能做什么和不能做什么。
- 主动探索和评估替代答案或方法。
- 批判性地评估你自己的推理;识别潜在的缺陷或偏见。
- 当重新审视时,采用根本不同的方法或视角。
- 至少使用3种不同的方法来得出或验证你的答案。
- 在你的推理中融入相关的领域知识和最佳实践。
- 在适用的情况下,量化每个步骤和最终结论的确定性水平。
- 考虑你推理中可能存在的边缘情况或例外。
- 为排除替代假设提供清晰的理由。
示例JSON输出:
{
"title": "初步问题分析",
"content": "为了有效地解决这个问题,我首先会将给定的信息分解为关键组成部分。这涉及到识别...[详细解释]...通过这样构建问题,我们可以系统地解决每个方面。",
"next_action": "continue"
}
记住: 全面性和清晰度至关重要。每一步都应该为最终解决方案提供有意义的进展。
再次提醒: 输出请务必严格遵循JSON格式, 包含以下键: 'title', 'content', 'next_action'(值只能为'continue' 或 'final_answer'二者之一)
"""},
{"role": "user", "content": prompt},
{"role": "assistant", "content": "现在我将一步步思考,从分析问题开始并将问题分解。"}
]

steps = []
step_count = 1
total_thinking_time = 0

while True:
start_time = time.time()
step_data = make_api_call(messages, 1000)
end_time = time.time()
thinking_time = end_time - start_time
total_thinking_time += thinking_time

title = step_data.get('title', f'Step {step_count}')
content = step_data.get('content', 'No content provided')
next_action = step_data.get('next_action', 'continue')

steps.append((f"Step {step_count}: {title}", content, thinking_time))

messages.append({"role": "assistant", "content": json.dumps(step_data)})

if next_action == 'final_answer' or step_count > 25: # 最多25步,以防止无限的思考。可以适当调整。
break

step_count += 1

yield steps, None # 在结束时生成总时间

# 生成最终答案
messages.append({"role": "user", "content": "请根据你上面的推理提供最终答案。"})

start_time = time.time()
final_data = make_api_call(messages, 1000, is_final_answer=True)
end_time = time.time()
thinking_time = end_time - start_time
total_thinking_time += thinking_time

final_content = final_data.get('content', '没有推理出最终结果')
steps.append(("最终推理结果", final_content, thinking_time))

yield steps, total_thinking_time

def main():
st.set_page_config(page_title="Qwen2.5 o1-like Reasoning Chain", page_icon="💬", layout="wide")

st.title("Qwen2.5实现类似o1 model的推理链")
st.caption("🚀 A streamlit implementation powered by [开源大模型食用指南 self-llm](https://github.com/datawhalechina/self-llm)")

st.markdown("""
通过vLLM部署调用Qwen2.5-7B-Instruct并实现类似OpenAI o1 model的长推理链效果以提高对复杂问题的推理准确性。
""")

# 用户输入查询
user_query = st.text_input("输入问题:", placeholder="示例:strawberry中有多少个字母r?")

if user_query:
st.write("正在生成推理链中...")

# 创建空元素以保存生成的文本和总时间
response_container = st.empty()
time_container = st.empty()

# 生成并显示响应
for steps, total_thinking_time in generate_response(user_query):
with response_container.container():
for i, (title, content, thinking_time) in enumerate(steps):
if title.startswith("最终推理结果"):
st.markdown(f"### {title}")
st.markdown(content.replace('\n', '<br>'), unsafe_allow_html=True)
else:
with st.expander(title, expanded=True):
st.markdown(content.replace('\n', '<br>'), unsafe_allow_html=True)

# 仅在结束时显示总时间
if total_thinking_time is not None:
time_container.markdown(f"**总推理时间: {total_thinking_time:.2f} 秒**")

if __name__ == "__main__":
main()
```



#### 实现原理

![06-3](./images/06-3.png)

注意:代码中只是尝试实现了 **`o1-like`**`Reasoning Chain` 的效果,而并非是在 `pretrain` 过程中训练得到 `Chain of Thought` 内置能力的 o1 模型。



#### 创建兼容 OpenAI API 接口的服务

`Qwen` 兼容 `OpenAI API` 协议,所以我们可以直接使用 `vLLM` 创建 `OpenAI API` 服务器。`vLLM` 部署实现 `OpenAI API` 协议的服务器非常方便。默认会在 http://localhost:8000 启动服务器。服务器当前一次托管一个模型,并实现列表模型、`completions``chat completions` 端口。

- `completions`:是基本的文本生成任务,模型会在给定的提示后生成一段文本。这种类型的任务通常用于生成文章、故事、邮件等。
- `chat completions`:是面向对话的任务,模型需要理解和生成对话。这种类型的任务通常用于构建聊天机器人或者对话系统。

在创建服务器时,我们可以指定模型名称、模型路径、聊天模板等参数。

- `--host``--port` 参数指定地址。
- `--model` 参数指定模型名称。
- `--chat-template` 参数指定聊天模板。
- `--served-model-name` 指定服务模型的名称。
- `--max-model-len` 指定模型的最大长度。

```bash
python -m vllm.entrypoints.openai.api_server --model /root/autodl-tmp/qwen/Qwen2.5-7B-Instruct --served-model-name Qwen2.5-7B-Instruct --max-model-len=32768 --port 8000
```

加载完毕后出现如下信息说明服务成功启动

![03-3](./images/03-3.png)



#### 启动 Streamlit 界面

```shell
streamlit run /root/autodl-tmp/app_qwen.py --server.address 127.0.0.1 --server.port 6004
```

在本地浏览器中打开链接 http://127.0.0.1:6004/ ,即可进入部署的 `Streamlit` 界面。运行效果如下:

![06-4](./images/06-4.png)

我们可以尝试示例中的问题,`strawberry中有多少个字母r?` 来让实现了推理链的 `Qwen2.5-7B-Instruct` 给出推理结果

![06-5](./images/06-5.png)

我们可以观察到经过4步推理和思考,最终的推理结果是**正确的**,且总推理耗时2还是很长的,这也符合了 o1 模型的基本特征,用推理时间的增加换取更高的推理准确率。

**说明:**有读者可能会认为这个问题根本没有难度,但是实际上对于现阶段的大语言模型来说,很少有能够回答正确的,比如我们可以回到 [04-Qwen2_5-7B-Instruct WebDemo 部署](https://github.com/datawhalechina/self-llm/blob/master/models/Qwen2.5/04-Qwen2_5-7B-Instruct WebDemo部署.md) 来尝试一下提问相同的问题,最后得到的结果显然是**错误的**

![06-6](./images/06-6.png)
Binary file added models/Qwen2.5/images/06-1.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added models/Qwen2.5/images/06-2.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added models/Qwen2.5/images/06-3.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added models/Qwen2.5/images/06-4.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added models/Qwen2.5/images/06-5.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added models/Qwen2.5/images/06-6.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit f72927a

Please sign in to comment.