模型简介
模型特点
模型能力
使用案例
库名称: transformers 许可证: apache-2.0 许可证链接: https://huggingface.co/Qwen/Qwen3-4B/blob/main/LICENSE 管道标签: 文本生成 基础模型: Qwen/Qwen3-4B
Qwen3-4B-AWQ
Qwen3亮点
Qwen3是通义千问系列最新一代大语言模型,提供全面的密集模型和混合专家(MoE)模型组合。基于大规模训练,Qwen3在推理、指令遵循、智能体能力和多语言支持方面取得突破性进展,主要特点包括:
- 独特支持思维模式(用于复杂逻辑推理、数学和编程)与非思维模式(用于高效通用对话)的单模型无缝切换,确保各类场景最优表现
- 显著增强推理能力,在数学、代码生成和常识逻辑推理方面超越前代QwQ(思维模式)和Qwen2.5指令模型(非思维模式)
- 卓越的人类偏好对齐,擅长创意写作、角色扮演、多轮对话和指令遵循,提供更自然、吸引人且沉浸式的对话体验
- 专业级智能体能力,能在思维和非思维模式下精准集成外部工具,在开源模型的复杂智能体任务中表现领先
- 支持100+种语言和方言,具备强大的多语言指令遵循和翻译能力
模型概览
Qwen3-4B具有以下特性:
- 类型:因果语言模型
- 训练阶段:预训练与后训练
- 参数量:40亿
- 非嵌入参数量:36亿
- 层数:36
- 注意力头数(GQA):查询头32个,键值头8个
- 上下文长度:原生支持32,768 tokens,通过YaRN扩展至131,072 tokens
- 量化方式:AWQ 4-bit
更多细节包括基准评估、硬件要求和推理性能,请参考我们的博客、GitHub和文档。
快速开始
Qwen3代码已集成至最新版Hugging Face transformers
,建议使用最新版本。
使用transformers<4.51.0
会遇到以下错误:
KeyError: 'qwen3'
以下代码片段展示如何基于给定输入生成内容:
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "Qwen/Qwen3-4B-AWQ"
# 加载分词器和模型
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:
# 反向查找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-4B-AWQ --reasoning-parser qwen3
- vLLM:
vllm serve Qwen/Qwen3-4B-AWQ --enable-reasoning --reasoning-parser deepseek_r1
更多使用指南请查看AWQ文档。
思维与非思维模式切换
[!TIP]
enable_thinking
开关同样适用于SGLang和vLLM创建的API。 请参考SGLang和vLLM用户文档。
enable_thinking=True
默认情况下Qwen3启用思维能力,类似QwQ-32B。这意味着模型将使用推理能力提升生成回复质量。例如在tokenizer.apply_chat_template
中显式设置enable_thinking=True
或保持默认值时,模型将进入思维模式。
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-4B-AWQ"):
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 = "strawberries中有多少个r?"
print(f"用户: {user_input_1}")
response_1 = chatbot.generate_response(user_input_1)
print(f"机器人: {response_1}")
print("----------------------")
# 带/no_think的第二次输入
user_input_2 = "那么blueberries中有多少个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-4B-AWQ',
# 使用阿里云模型服务平台端点:
# '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>回答内容`时
# # 不添加:当响应已被分离为thinking_content和content时
# 'thought_in_content': True,
# },
}
# 定义工具
tools = [
{'mcpServers': { # 可指定MCP配置文件
'time': {
'command': 'uvx',
'args': ['mcp-server-time', '--local-timezone=Asia/Shanghai']
},
"fetch": {
"command": "uvx",
"args": ["mcp-server-fetch"]
}
}
},
'code_interpreter', # 内置工具
]
# 定义智能体
bot = Assistant(llm=llm_cfg, function_list=tools)
# 流式生成
messages = [{'role': 'user', 'content': 'https://qwenlm.github.io/blog/ 介绍Qwen最新进展'}]
for responses in bot.run(messages=messages):
pass
print(responses)
处理长文本
Qwen3原生支持32,768 tokens的上下文长度。对于总长度(包括输入和输出)远超此限制的对话,建议使用RoPE缩放技术有效处理长文本。我们已验证模型通过YaRN方法在131,072 tokens上下文长度下的性能。
YaRN目前被多个推理框架支持,如本地使用的transformers
、部署用的vllm
和sglang
。通常有两种方式为支持的框架启用YaRN:
-
修改模型文件: 在
config.json
中添加rope_scaling
字段:{ ..., "rope_scaling": { "rope_type": "yarn", "factor": 4.0, "original_max_position_embeddings": 32768 } }
-
传递命令行参数:
对于
vllm
可使用:vllm serve ... --rope-scaling '{"rope_type":"yarn","factor":4.0,"original_max_position_embeddings":32768}' --max-model-len 131072
对于
sglang
可使用:python -


