ComfyUI-KJNodes
时间:2025/06/28
TorchCompileVAE
节点功能啊:使用 PyTorch 的 torch.compile() 功能来编译 VAE (变分自编码器) 模型,从而加速 VAE 的推理过程。
TorchCompileVAE-节点参数说明
输入参数 | |
vae | 输入待优化的 VAE 模型。 |
输出参数 | |
VAE | 编译优化后的 VAE 模型结构对象。 |
控件参数 | |
backend | 指定 torch.compile 的后端实现。 |
fullgraph | 启用 Full Graph 编译模式。启用后会尝试整体编译模型模块,提升运行效率,但失败率更高。建议逐步测试稳定性后再启用。 |
mode | 编译优化模式。 可选值: - default:推荐默认值,稳定。 - max-autotune:自动调优以获得最佳性能。 - reduce-overhead:减少开销,适用于低显存设备。 |
compile_encoder | 是否对编码器部分进行编译。建议开启,VAE 编码器通常负责将图像压缩为 latent 表示,计算密集,优化后能显著提升编码性能。 |
compile_decoder | 是否对解码器部分进行编译。建议开启,解码器将 latent 转换为图像,为最终输出环节,优化后可显著提升图像生成速度。 |