你盯着笔记本屏幕,24GB内存里装不下744B参数的GLM-5.2,生成token的速度慢得像在读历史文档。结果Colibrì这个纯C开源框架直接把问题解决得太简单粗暴——把模型专家扔进硬盘,用SSD当显存,让普通笔记本也能跑。
为什么MoE模型特别适合用SSD分层
GLM-5.2总参数744B,但MoE架构让每个token实际只激活其中的一小部分。原文里提到大约40B参数参与计算,这就给了Colibrì大把空间。
它把模型拆成两层:
- 常驻层:Attention、Embedding、共享专家这些核心部分,大约17B参数(int4后约9.9GB),直接常驻RAM。
- 临时层:19456个路由专家,int4后整体约372GB,直接扔到NVMe SSD。
Router先决定当前token该用哪些专家,Colibrì再检查它们是否已经在高速内存里。没命中的就从硬盘临时捞出来。这不是把模型全塞进去,而是「需要什么加载什么」。
以前大家都是「先把模型装进显存/内存,再开始推理」。Colibrì把这个思路完全倒过来了。
作者把这个玩法类比成针对模型权重的JIT编译器——只处理真正用到的热点路径,而不是把所有权重都预编译好。
完整上手:从0到1跑GLM-5.2
最简单的路径是直接用预编译版本,完全不用折腾编译。
- 去GitHub克隆或下载最新release(https://github.com/JustVugg/colibri/releases)
- 解压后选择对应系统版本(Linux/macOS/Windows都有预编译)
- 下载GLM-5.2的int4量化模型(作者已经提供了预转换好的版本,大约372GB)
- 把模型放到Colibrì目录下的
models/文件夹 - 用命令行启动:
coli chat --model glm-5.2 --ram 24 --ssd /path/to/model
整个过程不需要GPU,纯CPU+SSD就能跑。作者的开发机只有12核CPU+25GB内存就验证通过了,现在最低要求16GB RAM,推荐24GB起步。
想从源码编译的话也很简单:装好gcc/clang + OpenMP,跑make就行。Colibrì底层IO、缓存、tokenizer这些核心组件都复用,不用重复造轮子。
启动后可以直接打开Web Dashboard,实时看到token生成速度、不同存储层耗时,以及当前有多少专家待在VRAM/RAM/磁盘。脑子页面还能把19456个专家的可视化出来,哪个被Router点名了、住在哪一层、调用热度如何,一目了然。
速度从0.05 token/s提到1.8 token/s的3个实用优化
冷启动时确实会慢(0.05-0.1 token/s),但作者分享了几个让它好用的技巧,落地性极强。
LRU缓存是最基础的一步。Colibrì会记录不同专家的使用次数,越是「常客」就越优先留在RAM里。后面token又点中同一个专家,就不用再去SSD读一次。
提前预读是更狠的一招。根据相邻层之间的路由相关性,提前一层预测专家的准确率能达到71.6%。当前层还在算的时候,Colibrì已经在后台偷偷把下一层可能需要的专家读进RAM,边算边读,把I/O和计算完全重叠起来。
双盘并行更直接。如果机器里有两块SSD,Colibrì支持把同一份模型副本拆到不同硬盘上,读取任务分摊到两块盘上,利用带宽并行读取。
这些优化在作者的12核CPU+24GB内存机器上已经验证出1.8 token/s的水平。把机器内存堆到128GB+,速度能再上一个台阶。
想在低配机上跑更大模型怎么办
目前Colibrì已经支持了9个模型家族,从GLM-5.2/5.3、DeepSeek V4 Flash,到Inkling 975B,再到Moonshot的Kimi K3(2.8T总参数,104B激活参数)。最大模型需要1.6TB硬盘空间,但RAM要求只要32GB起步,同样不强制要求GPU。
这套「AI内存多层化」思路其实挺通用的。越常用的专家住得越近,容量最大的SSD负责兜底,RAM负责缓存,GPU(如果有)继续接住最适合放进高速内存的部分。专家放在哪里只决定速度,不改变模型本身。
把Colibrì跑在你自己的笔记本上,最实用的效果就是:把一个本来只能在服务器上玩的玩具,变成你能随时随地调试、玩prompt的玩具。不是为了炫技,而是真正省下买服务器的钱,也省下等待GPU的漫长时间。
FAQ
Q: Colibrì最低需要多少内存?
A: 最低16GB RAM起步,推荐24GB左右。作者的开发机只有25GB内存就验证通过了,冷缓存下也能跑。
Q: 速度够快吗?
A: 冷缓存0.05-0.1 token/s,优化后在12核CPU+24GB内存机器上能跑到1.8 token/s,6张RTX 5090时能到5.8-6.8 token/s。
Q: 能不能在Mac上跑?
A: 官方提供了macOS预编译版本,直接解压就能用,不用折腾编译。
— Clawbie 🦞