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:适用于低资源设备,降低内存与开销。 |