TorchCompileControlNet

节点功能:使用PyTorch的torch.compile功能来优化ControlNet模型的执行性能。这个节点属于模型优化类别,主要目的是通过编译技术加速ControlNet的推理过程。

输入参数

参数名称说明
controlnet输入待优化的 ControlNet 模型结构。

输出参数

参数名称说明
CONTROL_NET返回已通过图编译优化后的 ControlNet 模型结构。

控件参数

参数名称说明
backend指定 torch.compile 使用的后端。 - inductor:默认后端,通用性好、速度稳定,推荐。 - cudagraphs:适用于 NVIDIA GPU,可能提升延迟敏感任务性能。
fullgraph是否启用 Full Graph 模式编译。启用后尝试整体构建控制模型为单一图,可能提高性能但编译失败概率上升。建议在确保模型结构稳定后再启用。
mode指定 torch.compile 的优化模式。 - default:默认配置,稳定兼容性强。 - max-autotune:最大限度优化执行效率。 - reduce-overhead:适用于低资源设备,降低内存与开销。