
大型扩散模型不再只能在二三十GB显存的高端设备上顺畅运行。Hugging Face 7月23日把Nunchaku Lite原生接入Diffusers:开发者可以像加载普通模型一样调用4-bit检查点,不再单独安装推理引擎或在本机编译CUDA内核。在官方特定测试中,峰值显存可从31.1GB降至16.0GB;配合torch.compile,端到端耗时从3.00秒缩短到1.68秒。这次变化的价值不只是“压缩”,而是把量化模型真正放进主流工具链。
加载方式统一。预量化模型可直接通过Diffusers的from_pretrained()调用,CUDA内核首次使用时由kernels包获取。
权重和激活同时降到4-bit。它不是只省模型存储,而是让主要Transformer层以W4A4执行,兼顾显存和去噪速度。
新架构可自行适配。配套工具能完成校准、量化、打包和发布,开发者不必等待每个模型都有专用引擎。
扩散Transformer直接做4-bit量化并不轻松。权重和激活中都可能出现幅度很大的异常值,粗暴压缩会放大量化误差,最终表现为细节损失、颜色漂移或结构不稳定。Nunchaku背后的SVDQuant先把激活中的异常值迁移到权重,再用奇异值分解抽出一条小型16-bit低秩支路;剩余的大部分计算进入4-bit支路。这样,最难压缩的信息保留更高精度,主体计算仍能享受低位宽优势。
原版Nunchaku依靠针对模型结构定制的融合算子,把低秩支路与4-bit矩阵乘合并,速度更激进,但每增加一个架构都要做专门适配。Nunchaku Lite选择更通用的路线:运行时把标准Diffusers模型中的线性层替换为SVDQ W4A4或AWQ W4A16层,模型结构和常用加载钩子仍保持兼容。代价是它无法自动复现所有QKV、归一化和旋转位置编码融合,因此官方把基础加速描述为约30%,而不是直接套用专用引擎的更高数字。
官方基准使用RTX PRO 6000 Blackwell,在1024×1024分辨率下运行ERNIE-Image-Turbo。BF16基线端到端耗时3.00秒、峰值显存31.1GB;Nunchaku Lite NVFP4为2.27秒和20.6GB,约1.35倍速度。加入torch.compile后耗时降至1.68秒,即1.8倍,但显存仍为20.6GB。另一组把文本编码器继续量化为NF4后,显存降至16.0GB,耗时为2.29秒。也就是说,“显存减半”和“1.8倍”来自两种优化组合,不能简单视为同一次运行同时达到的结果。
硬件范围也有限。NVFP4检查点需要Blackwell GPU,包括RTX 50系列、RTX PRO 6000和B200;较早设备可使用INT4路径,覆盖Turing、Ampere和Ada。当前4-bit内核不支持Volta与Hopper,量化器会在加载阶段检查CUDA能力。官方还强调图像质量接近BF16,但公开页面主要给出同种子样图,没有覆盖更多模型、提示词、分辨率和长批次的独立统计,生产部署仍应做自己的回归测试。
对个人开发者和小团队,最直接的收益是减少为了显存而频繁CPU卸载,降低单机运行大模型的等待时间。对模型发布者,量化配置可以随普通Diffusers仓库一起分发,用户仍能使用调度器、LoRA、卸载和编译等既有能力。配套的diffuse-compressor还提供从扫描目标层、校准到生成safetensors检查点的完整流程,使新图像架构更快获得低显存版本。
但通用并不等于零成本。不同架构的调制层、融合投影和文本编码器仍需识别,某些结构改写必须提供目标配置或运行时适配器。量化也不能替代模型许可证、GPU兼容性和输出质量检查。对于视频或更高分辨率任务,VAE、文本编码器、注意力缓存等组件还会占用额外内存,单个图像基准不能直接外推到所有生成工作流。
ATYUN编辑判断:Nunchaku Lite最值得关注的不是刷新一张性能表,而是把“权重与激活同时4-bit”从独立引擎能力,变成Diffusers中的标准加载路径。它让开发者可以先用预量化检查点验证收益,再决定是否投入专用融合优化,显著降低试错门槛。短期内,它更像一套实用的工程桥梁,而非对所有扩散模型都成立的万能加速器;其长期价值取决于更多架构、GPU世代和第三方画质评测能否跟上。
