ComfyUI Flux Accelerator:加速 Flux.1 模型图像生成的ComfyUI自定义节点

ComfyUI Flux Accelerator是一个能够显著加速 Flux.1 模型图像生成的一个ComfyUI自定义节点,开发者号称可实现35%-37% 的加速效果,不过可惜的是目前仅支持Linux,还会损失一些图像的生成质量。

ComfyUI Flux Accelerator 如何加速图像生成?

  • 利用 TAEF1,这是一种快速高效的自动编码器,能够在极短的时间内对像素进行编码和解码,尽管这会略微降低图像质量。
  • 通过量化和编译,ComfyUI Flux Accelerator 使用 torchao 和 torch.compile() 对模型进行优化,提高其运行速度。
  • 跳过冗余的 DiT 块(DiT blocks),这一选项可以提高生成速度,用户可以在节点中选择跳过块的数量(默认为 3,即 MMDiT 块的 12 个)。

ComfyUI Flux Accelerator 能提升多少速度?

使用 ComfyUI Flux Accelerator,图像生成速度可以比默认设置快达 37.25%。以下是在 RTX 4090 上的测试示例:

  • 512x512 4steps: 从 0.51 秒缩短到 0.32 秒(提速 37.25%)
  • 1024x1024 4steps: 1.94s → 1.24s (36.08% faster)
  • 1024x1024 20step: 8.77s → 5.74s (34.55% faster)

如何使用 ComfyUI Flux Accelerator?

只需在工作流程中加入 FluxAccelerator 节点即可。如果你的显卡显存少于 24GB,在调整参数时可能会遇到内存不足的错误(Out Of Memory errors)。遇到这种情况,只需忽略错误并重新运行即可。

ComfyUI Flux Accelerator 的局限性:

  • 图像质量:为了提升速度,ComfyUI Flux Accelerator 会牺牲一小部分图像质量。
  • 编译时间:首次编译模型可能需要 30-60 秒,因为它使用了 torch.compile() 来优化模型。
  • 兼容性:目前,ComfyUI Flux Accelerator 仅支持 Linux 系统。
0

评论0

没有账号?注册  忘记密码?