TorchCompileControlNet

El nodo TorchCompileControlNet, que internamente utiliza torch.compile.

TorchCompileControlNet es un nodo de utilidad diseñado para optimizar el rendimiento de inferencia de los modelos ControlNet. Aprovecha la tecnología de compilación justo a tiempo (JIT) de PyTorch para convertir el modelo ControlNet en un formato de ejecución más eficiente, con el objetivo de acelerar el flujo de trabajo de generación de imágenes.

Funcionalidad del Nodo

Este nodo utiliza internamente la función torch.compile para compilar el modelo ControlNet de entrada. Permite a los usuarios seleccionar diferentes backends de compilación (como inductor o cudagraphs) y modos de optimización (como max-autotune), y admite la habilitación del modo de gráfico completo para optimizaciones más agresivas. El proceso de compilación es único; el nodo registra internamente el estado de compilación para evitar operaciones repetidas.

Explicación de los Parámetros del Nodo - TorchCompileControlNet

Entradas (Inputs)

Nombre del Parámetro	Tipo de Dato	Requerido	Valor por Defecto	Rango/Opción	Descripción
`controlnet`	CONTROL_NET	Sí	-	-	: Introduce el modelo ControlNet que se va a compilar y optimizar.

Parámetros de Control (Parameters)

Nombre del Parámetro	Tipo de Dato	Requerido	Valor por Defecto	Rango/Opción	Descripción
`backend`	COMBO	Sí	inductor	"inductor", "cudagraphs"	: Selecciona el motor backend utilizado por la compilación de PyTorch. Los valores posibles son `inductor` o `cudagraphs`.
`mode`	COMBO	Sí	default	"default", "max-autotune", "max-autotune-no-cud...	: Selecciona el modo de optimización para la compilación. Las opciones son `default`, `max-autotune`, `max-autotune-no-cudagraphs` o `reduce-overhead`.
`fullgraph`	BOOLEAN	Sí	False	-	Habilita el modo gráfico completo

Salidas (Outputs)

Nombre del Parámetro	Tipo de Dato	Descripción
CONTROL_NET	CONTROL_NET	: Devuelve el modelo ControlNet compilado y optimizado.

Casos de Uso

En flujos de trabajo que requieren iteración rápida o generación por lotes de imágenes, puedes conectar este nodo después del cargador de ControlNet y antes del muestreador (sampler). Al compilar el ControlNet, se puede reducir significativamente el tiempo consumido por una o múltiples muestras, manteniendo la calidad de la generación. Esto es especialmente útil para aplicaciones en tiempo real que exigen velocidad o para escenarios de creación de contenido a gran escala.

Consideraciones

Este nodo está marcado como una característica experimental. Su efectividad de compilación y estabilidad pueden variar según la versión de PyTorch, los controladores de hardware y el modelo específico. Si la compilación falla, el nodo lanzará un error de tiempo de ejecución. Deberás verificar la compatibilidad del entorno o probar diferentes combinaciones de parámetros de compilación.

Enlace al Código Fuente del Nodo TorchCompileControlNet

El nodo TorchCompileControlNet proviene del paquete de nodos ComfyUI-KJNodes.