从 通义千问 官方 GitHub README 翻译整理的快速上手指南。
通义千问(Qwen)是阿里云开源的大语言模型系列,包含基础语言模型(Qwen)和对话模型(Qwen-Chat)两大类。目前提供四个尺寸版本:1.8B、7B、14B、72B,参数规模从18亿到720亿不等,满足不同场景和硬件需求。
每个版本还提供 Int4 和 Int8 量化版本,可在更低显存下运行。模型已在 Hugging Face 和 ModelScope 平台开源,并发布了配套的技术报告。
⚠️ 注意:该仓库(QwenLM/Qwen)已不再积极维护,最新版本请关注 QwenLM/Qwen2。
pip install transformers accelerate
如需使用量化版本,额外安装:
pip install auto-gptq optimum
国内用户推荐从 ModelScope 下载:
from modelscope import snapshot_download
model_dir = snapshot_download('qwen/Qwen-7B-Chat')
from transformers import AutoModelForCausalLM, AutoTokenizer
from modelscope import snapshot_download
# 下载并加载模型
model_dir = snapshot_download('qwen/Qwen-7B-Chat')
tokenizer = AutoTokenizer.from_pretrained(model_dir, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
model_dir,
device_map="auto",
trust_remote_code=True
).eval()
# 对话
response, history = model.chat(tokenizer, "你好,请介绍一下你自己", history=None)
print(response)
history = []
while True:
query = input("用户:")
if query == "exit":
break
response, history = model.chat(tokenizer, query, history=history)
print(f"Qwen:{response}")
# 加载 Int4 版本
model_dir = snapshot_download('qwen/Qwen-7B-Chat-Int4')
model = AutoModelForCausalLM.from_pretrained(
model_dir,
device_map="auto",
trust_remote_code=True,
use_quantization=True
).eval()
| 资源 | 链接 |
|---|---|
| 模型下载(Hugging Face) | 🤗 Qwen |
| 模型下载(ModelScope) | 🤖 Qwen |
| 在线体验 Demo | 🖥️ 72B-Chat 演示 |
| 技术报告 | 📑 arXiv 论文 |
| 官方 API | DashScope |
| 微信交流 | 扫码见仓库 assets/wechat.png |
| Discord 社区 | 加入 Discord |
翻译整理自 GitHub README,原文:https://github.com/QwenLM/Qwen