跳到正文
IFAI

AI工具

NVIDIA 上线 FLUX.1 Kontext NIM 微服务:显存降至 7GB,速度翻倍

首次发布 传播指数 38

NVIDIA 推出 FLUX.1 Kontext [dev] NIM 微服务。通过 FP8 与 FP4 量化,模型占用从 24GB 最低压缩至 7GB,结合 TensorRT 带来超 2 倍推理加速,且支持在 ComfyUI 中一键调用。

图:blogs.nvidia.com

想在本地用开源大模型改图,最让人头疼的往往不是显卡不够贵,而是装环境和显存爆炸。

部署一个几十 GB 的大模型,既要把模型文件转成推理引擎支持的格式,又得自己折腾量化压缩显存,最后还得配环境接 API。折腾大半天,显存可能还是撑不住。

NVIDIA 和 Black Forest Labs 刚刚把这套门槛拆了。Black Forest Labs 的图像编辑模型 FLUX.1 Kontext [dev] 正式上线了 NVIDIA NIM 微服务,专门给本地 RTX AI PC 做了深度适配。

24GB 压到 7GB,显存减负与两倍提速

FLUX.1 Kontext [dev] 本身是一个支持文本和图像双输入的开源权重模型。用户只要给一张图,打几句日常的大白话,就能指挥它微调局部细节或者大改整个场景,省去了复杂的微调或工作流搭建。

这次双方合作的核心,在于直接给模型动了手术。原始的 BF16 模型足足有 24GB,一般消费级显卡根本吃不消。

NVIDIA 针对不同架构的显卡做了针对性量化。面向 GeForce RTX 40 系列等 Ada 架构显卡,他们推出了 FP8 版本,直接调用 Tensor Core 的 FP8 加速能力,把体积腰斩到 12GB。

面向最新的 GeForce RTX 50 系列等 Blackwell 架构显卡,官方采用了一种名为 SVDQuant 的新方法,压缩出只有 7GB 的 FP4 版本,在缩小体积的同时保住了画面质量。

除了体积缩减,NVIDIA 还拿出了底层的 TensorRT 框架,充分调用显卡里的 Tensor Core。相比在 PyTorch 框架下跑原始 BF16 模型,推理速度直接翻了一倍以上。

以往要想吃满这些性能红利,往往需要专业的 AI 基础设施开发经验。但现在通过 ComfyUI 的 NIM 节点,预打包的文件支持一键下载,普通玩家也能直接享受到这种速度提升。

想在本地跑起来也不算繁琐。用户需要先装好 NVIDIA AI Workbench 和 ComfyUI,在软件内通过 ComfyUI Manager 安装 NIM 节点,去 Hugging Face 页面同意 FLUX.1 Kontext [dev] 的许可协议,随后在工作流中点一下运行,插件就会自动帮你把需要的模型和流程拉取完整。

为什么重要

普通创作者要在本地流畅编辑高清大图,以往要么得买昂贵的工作站显卡,要么就得忍受云端收费与排队。现在借助 FP8 和 FP4 两个量化版本,12GB 显存的 40 系列显卡就能轻松跑起来,未来的 50 系列更是只要 7GB 显存,本地修图的门槛被大幅拉低。

我们认为,NVIDIA 这一步走得相当精明。它没有只卖硬件算力,而是把复杂的底层加速、格式转换和量化打包成了即开即用的 NIM 微服务,直接缝合进了创作者最扎堆的 ComfyUI 生态中。这种做法让普通爱好者不用懂底层代码,就能榨干显卡的每一分算力。

看起来,这种软硬件深度绑定的打法,也会继续加深 NVIDIA 在端侧 AI 创作工具链上的护城河。当开源模型发布的第一时间就绑定了私有的 TensorRT 优化与专用微服务,竞争对手不仅要在硬件能效上追赶,在软件易用性和工作流集成上也会被甩开更远。

不过隐患依然存在。用户必须严格依赖特定的架构代际才能获得最优性能,比如 FP4 的极致压缩仅留给了 Blackwell 架构,这种硬件分代策略在实际体验中可能会把老卡用户隔在门外。

来源 (1)

  1. [1]
    blogs.nvidia.com(在新标签页打开)原文blogs.nvidia.com

每日 AI 内容精选