ComfyUI-KJNodes
时间:2025/06/28
TorchCompileControlNet
节点功能:使用PyTorch的torch.compile功能来优化ControlNet模型的执行性能。这个节点属于模型优化类别,主要目的是通过编译技术加速ControlNet的推理过程。
TorchCompileControlNet-节点参数说明
输入参数 | |
controlnet | 输入待优化的 ControlNet 模型结构。 |
输出参数 | |
CONTROL_NET | 返回已通过图编译优化后的 ControlNet 模型结构。 |
控件参数 | |
backend | 指定 torch.compile 使用的后端。 - inductor:默认后端,通用性好、速度稳定,推荐。 - cudagraphs:适用于 NVIDIA GPU,可能提升延迟敏感任务性能。 |
fullgraph | 是否启用 Full Graph 模式编译。启用后尝试整体构建控制模型为单一图,可能提高性能但编译失败概率上升。建议在确保模型结构稳定后再启用。 |
mode | 指定 torch.compile 的优化模式。 - default:默认配置,稳定兼容性强。 - max-autotune:最大限度优化执行效率。 - reduce-overhead:适用于低资源设备,降低内存与开销。 |