首页/AI/大模型部署与运维速查/bitsandbytes:即用量化(int8/NF4)
本包目录(267 条)
部署形态与选型43 条推理环境与 GPU50 条
推理参数与优化50 条
显存与并发估算40 条服务化与网关34 条运维监控与排障50 条

bitsandbytes:即用量化(int8/NF4)

bitsandbytes 在加载时做 8 位或 4 位(NF4)量化,无需预量化权重:上手快,常用于实验与显存紧张的单机场景;推理吞吐通常不及面向推理优化的量化内核。

4 位加载(示意)

4 位加载(示意)
from transformers import AutoModelForCausalLM, BitsAndBytesConfig
bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_quant_type="nf4")
model = AutoModelForCausalLM.from_pretrained("<模型路径>", quantization_config=bnb, device_map="cuda")
NF4 与 int8 都属即用量化:反量化有开销,做在线高并发前先与 AWQ/GPTQ/FP8 路线对比吞吐。

易错点

  • 拿 NF4 直接跑在线高并发:反量化开销随批量放大,吞吐可能明显低于离线量化权重

数据来源与参考

  • Hugging Face bitsandbytes 量化文档(huggingface.co/docs/transformers)
  • bitsandbytes 仓库(github.com/bitsandbytes-foundation/bitsandbytes)
​‌‌​​‌​​​‌‌​‌‌‌‌​‌‌​​​‌‌​‌‌‌​​‌‌​​‌​‌‌‌​​‌‌‌​​​‌​‌‌​‌​​‌​‌‌‌​​​‌​‌‌​‌​​‌​‌‌​​​‌‌​‌‌‌​‌​​​‌‌​‌‌‌‌​​‌​‌‌‌​​‌‌​​​‌‌​‌‌​‌‌‌‌​‌‌​‌‌​‌