Skip to content
简体中文

切换主题

// 零一万物 Yi · 文本大模型

Yi-Coder-9B-Chat 本地部署与显存需求

8.83B 参数。4 位量化约 7.2 GB,BF16 约 20.7 GB(1 个 8K token 请求)。

发布 2024-08-21许可:可商用显存为估算值

01规格

发布方零一万物 · 01-ai
系列零一万物 Yi · Yi-Coder
类型文本大模型 (text-generation)
参数量8.83B 来源:按官方 safetensors 权重统计
上下文长度128K tokens (131,072)取自 config.json 的 max_position_embeddings
结构48 层 · GQA
发布权重BF16 · 17.7 GB下载量
许可Apache-2.0 · 可商用宽松许可(Apache-2.0 / MIT / CC-BY / OpenMDW):允许商用。
发布日期2024-08-21 (Hugging Face 仓库创建日)

02需要多少显存(估算)

合计 = 权重 + 10% + 0.5GB 运行开销 + 1 个请求的 KV 缓存(FP16)。并发用户越多,KV 部分成倍增加。

精度权重合计 @ 8K合计 @ 32K合计 @ 128K最低显卡Mac
BF16 / FP16(原始精度)17.7 GB 20.7 GB23.1 GB32.8 GB RTX 4090 24GBMac 36GB
FP8 / INT8 / Q8_09.4 GB 11.6 GB14.0 GB23.7 GB 16GB 显卡Mac 24GB
4 位量化(Q4_K_M / AWQ / GPTQ-Int4)5.3 GB 7.2 GB9.6 GB19.3 GB 8GB 显卡Mac 16GB

每个请求的 KV 缓存:8K 0.8 GB · 32K 3.2 GB · 128K 12.9 GB

每权重比特:BF16 16、FP8 8.5、4 位 4.85。“放得下” = 合计 ≤ 显存 90%;Mac GPU 约可用统一内存的 2/3(≤36GB)或 3/4(更大机型)。MoE 模型所有专家都要进显存,按总参数算。 估算方法 →

03怎么跑

只列厂商或推理框架官方明确支持的方式。

ollama run yi-coder:9b

怎么选推理引擎 →

04官方仓库与量化版

01-ai 官方发布的 2 个仓库。

05零一万物 Yi 其他模型

全部 9 个 →
模型参数4 位合计发布
Yi-Coder-1.5B-Chat · 同系列1.48B3.1 GB2024-08-21
Yi-1.5-9B-Chat-16K8.83B7.2 GB2024-05-15
Yi-1.5-34B-Chat-16K34.4B25.4 GB2024-05-15
Yi-1.5-9B-32K8.83B7.2 GB2024-05-15
Yi-1.5-34B-32K34.4B25.4 GB2024-05-15
Yi-1.5-6B6.06B5.1 GB2024-05-11
Yi-1.5-9B-Chat8.83B7.2 GB2024-05-10
Yi-1.5-34B-Chat34.4B25.4 GB2024-05-10

06来源

数据核对于 2024-09-12。参数、上下文、许可取自官方仓库;显存为本站估算。

数据更新于:自建算力 · 开源模型(北京时间)