TorchCompileControlNet

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