模型简介
模型特点
模型能力
使用案例
库名称: transformers
许可证: apache-2.0
许可证链接: https://huggingface.co/Qwen/Qwen3-0.6B/blob/main/LICENSE
流水线标签: 文本生成
基础模型:
- Qwen/Qwen3-0.6B-Base
QuantFactory/Qwen3-0.6B-GGUF
这是使用 llama.cpp 创建的 Qwen/Qwen3-0.6B 量化版本
原始模型卡片
Qwen3-0.6B
Qwen3 亮点
Qwen3 是 Qwen 系列最新一代的大语言模型,提供了一系列密集型和混合专家(MoE)模型。基于广泛的训练,Qwen3 在推理、指令遵循、代理能力和多语言支持方面取得了突破性进展,具有以下关键特点:
- 独特支持在单一模型内无缝切换思维模式(用于复杂逻辑推理、数学和编码)和非思维模式(用于高效的通用对话),确保在各种场景下的最佳性能。
- 显著提升推理能力,在数学、代码生成和常识逻辑推理方面超越了之前的 QwQ(思维模式)和 Qwen2.5 指令模型(非思维模式)。
- 卓越的人类偏好对齐,在创意写作、角色扮演、多轮对话和指令遵循方面表现出色,提供更自然、引人入胜和沉浸式的对话体验。
- 擅长代理能力,能够在思维和非思维模式下精确集成外部工具,并在复杂的基于代理的任务中达到开源模型的领先性能。
- 支持 100 多种语言和方言,具备强大的多语言指令遵循和翻译能力。
模型概览
Qwen3-0.6B 具有以下特点:
- 类型:因果语言模型
- 训练阶段:预训练和后训练
- 参数数量:0.6B
- 非嵌入参数数量:0.44B
- 层数:28
- 注意力头数(GQA):16 个 Q 头和 8 个 KV 头
- 上下文长度:32,768
更多细节,包括基准评估、硬件要求和推理性能,请参考我们的博客、GitHub 和文档。
[!TIP]
如果遇到严重的无限重复问题,请参考最佳实践部分获取最佳采样参数,并将presence_penalty
设置为 1.5。
快速开始
Qwen3 的代码已集成到最新版本的 Hugging Face transformers
中,建议使用最新版本的 transformers
。
如果使用 transformers<4.51.0
,会遇到以下错误:
KeyError: 'qwen3'
以下代码片段展示了如何基于给定输入使用模型生成内容:
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "Qwen/Qwen3-0.6B"
# 加载分词器和模型
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype="auto",
device_map="auto"
)
# 准备模型输入
prompt = "给我一个关于大语言模型的简短介绍。"
messages = [
{"role": "user", "content": prompt}
]
text = tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True,
enable_thinking=True # 在思维和非思维模式之间切换。默认为 True。
)
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
# 进行文本补全
generated_ids = model.generate(
**model_inputs,
max_new_tokens=32768
)
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
# 解析思维内容
try:
# rindex 查找 151668 (</think>)
index = len(output_ids) - output_ids[::-1].index(151668)
except ValueError:
index = 0
thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("\n")
content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("\n")
print("思维内容:", thinking_content)
print("内容:", content)
对于部署,可以使用 sglang>=0.4.6.post1
或 vllm>=0.8.5
创建兼容 OpenAI 的 API 端点:
- SGLang:
python -m sglang.launch_server --model-path Qwen/Qwen3-0.6B --reasoning-parser qwen3
- vLLM:
vllm serve Qwen/Qwen3-0.6B --enable-reasoning --reasoning-parser deepseek_r1
对于本地使用,Ollama、LMStudio、MLX-LM、llama.cpp 和 KTransformers 等应用程序也已支持 Qwen3。
在思维和非思维模式之间切换
[!TIP]
enable_thinking
开关也可用于 SGLang 和 vLLM 创建的 API。
请参考我们的文档了解 SGLang 和 vLLM 用户。
enable_thinking=True
默认情况下,Qwen3 启用了思维能力,类似于 QwQ-32B。这意味着模型将使用其推理能力来提升生成响应的质量。例如,当显式设置 enable_thinking=True
或在 tokenizer.apply_chat_template
中保留默认值时,模型将进入思维模式。
text = tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True,
enable_thinking=True # enable_thinking 的默认值为 True
)
在此模式下,模型将生成包裹在 <think>...</think>
块中的思维内容,随后是最终响应。
[!NOTE]
对于思维模式,使用Temperature=0.6
、TopP=0.95
、TopK=20
和MinP=0
(generation_config.json
中的默认设置)。不要使用贪婪解码,因为这会导致性能下降和无限重复。更多详细指导,请参考最佳实践部分。
enable_thinking=False
我们提供了一个硬开关来严格禁用模型的思维行为,使其功能与之前的 Qwen2.5-Instruct 模型保持一致。此模式在需要禁用思维以提高效率的场景中特别有用。
text = tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True,
enable_thinking=False # 设置 enable_thinking=False 禁用思维模式
)
在此模式下,模型不会生成任何思维内容,也不会包含 <think>...</think>
块。
[!NOTE]
对于非思维模式,建议使用Temperature=0.7
、TopP=0.8
、TopK=20
和MinP=0
。更多详细指导,请参考最佳实践部分。
高级用法:通过用户输入在思维和非思维模式之间切换
我们提供了一个软开关机制,允许用户在 enable_thinking=True
时动态控制模型的行为。具体来说,可以在用户提示或系统消息中添加 /think
和 /no_think
,以在多轮对话中逐轮切换模型的思维模式。模型将遵循最近一次指令。
以下是一个多轮对话的示例:
from transformers import AutoModelForCausalLM, AutoTokenizer
class QwenChatbot:
def __init__(self, model_name="Qwen/Qwen3-0.6B"):
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
self.model = AutoModelForCausalLM.from_pretrained(model_name)
self.history = []
def generate_response(self, user_input):
messages = self.history + [{"role": "user", "content": user_input}]
text = self.tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True
)
inputs = self.tokenizer(text, return_tensors="pt")
response_ids = self.model.generate(**inputs, max_new_tokens=32768)[0][len(inputs.input_ids[0]):].tolist()
response = self.tokenizer.decode(response_ids, skip_special_tokens=True)
# 更新历史
self.history.append({"role": "user", "content": user_input})
self.history.append({"role": "assistant", "content": response})
return response
# 示例用法
if __name__ == "__main__":
chatbot = QwenChatbot()
# 第一次输入(不带 /think 或 /no_think 标签,默认启用思维模式)
user_input_1 = "草莓中有多少个 r?"
print(f"用户: {user_input_1}")
response_1 = chatbot.generate_response(user_input_1)
print(f"机器人: {response_1}")
print("----------------------")
# 第二次输入带 /no_think
user_input_2 = "那么,蓝莓中有多少个 r? /no_think"
print(f"用户: {user_input_2}")
response_2 = chatbot.generate_response(user_input_2)
print(f"机器人: {response_2}")
print("----------------------")
# 第三次输入带 /think
user_input_3 = "真的吗? /think"
print(f"用户: {user_input_3}")
response_3 = chatbot.generate_response(user_input_3)
print(f"机器人: {response_3}")
[!NOTE]
为了 API 兼容性,当enable_thinking=True
时,无论用户使用/think
还是/no_think
,模型始终会输出包裹在<think>...</think>
块中的内容。然而,如果思维被禁用,该块中的内容可能为空。
当enable_thinking=False
时,软开关无效。无论用户输入任何/think
或/no_think
标签,模型都不会生成思维内容,也不会包含<think>...</think>
块。
代理使用
Qwen3 在工具调用能力方面表现出色。建议使用 Qwen-Agent 以充分利用 Qwen3 的代理能力。Qwen-Agent 内部封装了工具调用模板和工具调用解析器,大大降低了编码复杂度。
要定义可用工具,可以使用 MCP 配置文件、使用 Qwen-Agent 的集成工具,或自行集成其他工具。
from qwen_agent.agents import Assistant
# 定义 LLM
llm_cfg = {
'model': 'Qwen3-0.6B',
# 使用 Alibaba Model Studio 提供的端点:
# 'model_type': 'qwen_dashscope',
# 'api_key': os.getenv('DASHSCOPE_API_KEY'),
# 使用兼容 OpenAI API 的自定义端点:
'model_server': 'http://localhost:8000/v1', # api_base
'api_key': 'EMPTY',
# 其他参数:
# 'generate_cfg': {
# # 添加:当响应内容为 `<think>这是思考</think>这是答案` 时;
# #


