
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 16362
技术标签
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 环境与 pip(README 未指定具体版本)
- 已安装 PyTorch,模型推理依赖(技术栈含 pytorch)
- 已安装 huggingface-cli,用于下载模型权重
- 可用的 NVIDIA GPU(README 基准测试使用 RTX 4070 Laptop 8GB)
- 足够的磁盘空间存放模型权重(示例为 Qwen3-0.6B)
安装与启动步骤
-
1安装 nano-vllm
README 唯一给出的安装方式:直接从 GitHub 仓库通过 pip 安装,无需 clone 源码。
pip install git+https://github.com/GeeeekExplorer/nano-vllm.git -
2下载模型权重
用 huggingface-cli 把 Qwen3-0.6B 权重拉到本地目录,--resume-download 支持断点续传。
huggingface-cli download --resume-download Qwen/Qwen3-0.6B --local-dir ~/huggingface/Qwen3-0.6B/ --local-dir-use-symlinks False -
3编写调用脚本
按 README 示例导入 LLM 与 SamplingParams,把 /YOUR/MODEL/PATH 换成上一步的本地权重目录。
from nanovllm import LLM, SamplingParams llm = LLM("/YOUR/MODEL/PATH", enforce_eager=True, tensor_parallel_size=1) sampling_params = SamplingParams(temperature=0.6, max_tokens=256) prompts = ["Hello, Nano-vLLM."] outputs = llm.generate(prompts, sampling_params) outputs[0]["text"] -
4运行示例脚本
仓库内 example.py 已给出完整用法,确认其中模型路径为本地真实路径后直接执行。
python example.py
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
模型路径(LLM 第一个参数) | 是 | 指向本地模型权重目录,必须替换为真实路径 | ~/huggingface/Qwen3-0.6B/ |
enforce_eager | 是 | 示例中设为 True,关闭图捕获等激进优化以便运行 | True |
tensor_parallel_size | 是 | 张量并行份数,示例单卡设为 1 | 1 |
temperature | 否 | 采样温度,值越高输出越随机 | 0.6 |
max_tokens | 否 | 单条生成的最大 token 数 | 256 |
如何确认成功
脚本能正常打印 outputs[0]["text"] 的内容,即模型成功生成文本、推理流程跑通。
常见问题
Q:必须用 GPU 吗?
A:README 的基准测试跑在 RTX 4070 Laptop 8GB 上,未提供 CPU 运行说明,建议准备 NVIDIA GPU 环境。
Q:下载模型中断了怎么办?
A:README 命令带 --resume-download,可直接重跑同一条命令续传;也可指定其他本地目录避免重复下载。
Q:和 vLLM 的接口一样吗?
A:API 基本对齐 vLLM,主要差异在 LLM.generate 的用法,具体参考仓库中的 example.py。
Q:支持哪些模型?
A:README 示例使用 Qwen/Qwen3-0.6B,基准测试使用 Qwen3,其他模型需自行验证。
Q:项目有哪些优化特性?
A:README 列出前缀缓存、张量并行、Torch 编译、CUDA graph,以及连续批处理和 PagedAttention。
注意事项
- README 未给出 Python 版本与依赖清单,需自行准备匹配 PyTorch 的环境。
- 代码约 1200 行,适合作为学习 LLM 推理原理的参考实现。
- 复制示例代码时务必把 /YOUR/MODEL/PATH 改成自己的本地权重目录。
- 定位是快速离线推理(offline inference),README 未提供在线服务部署说明。
核心亮点
- 代码量少且结构清晰,适合快速上手和教学
- 实现了 PagedAttention 和连续批处理等核心优化
- 部署简单,资源占用远低于完整版 vLLM
不足之处
- 功能覆盖有限,缺少高级特性如量化、多模态支持
- 文档/社区待观察
适用场景
- 学习 LLM 推理原理和优化技术
- 边缘设备或资源受限环境下的模型推理
- 快速原型验证和教学演示
替代项目
vLLM、llama.cpp、TensorRT-LLM
上一篇:unstructured
下一篇:casdoor
同类项目推荐
pymilo
开源
统一序列化 ML 模型,跨框架安全保存与加载
PyMilo: Python for ML I/O
SharpVector
开源
在 .NET 应用内直接实现语义搜索,无需外部数据库。
Lightweight, In-memory, Semantic Search, Text Vector Database to embed in any .NET A···
SageAttention
开源
渲染提速好几倍,出片快人一步
[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of ···
awesome-ai-infrastructure
开源
一份清单,帮你快速选型AI基础设施工具,少走弯路。
A curated list of awesome tools, frameworks, platforms, and resources for building s···
llm-inference
开源
快速搭建基于开源大模型的聊天机器人,省去推理部署烦恼
Large Language Model (LLM) Inference API and Chatbot
stable-diffusion-rocm-docker
开源
AMD显卡用户,一条命令跑起Stable Diffusion
Stable Diffusion Docker image preconfigured for usage with AMD Radeon cards
corpusos
开源
一套协议打通所有 AI 框架,告别碎片化集成
Open-source protocol suite standardizing LLM, Vector, Graph, and Embedding infrastru···
mlplatform-workshop
开源
用示例代码快速搭建机器学习平台骨架
Example code for a basic ML Platform based on Pulumi, FastAPI, DVC, MLFlow and more
Memori
开源
给 AI 装上持久记忆,让智能体不再失忆
Memori is agent-native memory infrastructure. A LLM-agnostic layer that turns agent ···