VRAM Debug (VRAM_Debug) - ComfyUI-KJNodes

El nodo VRAM Debug obtiene el tamaño actual de la memoria de video (VRAM) disponible en bytes mediante la interfaz model_management de ComfyUI.

VRAM Debug es un nodo de utilidad para monitorizar y gestionar la memoria de video (VRAM). Su función principal es medir e informar de la capacidad de VRAM disponible durante la ejecución de un flujo de trabajo, ofreciendo además opciones para liberar VRAM y descargar modelos, ayudando así a los usuarios a optimizar el uso de recursos.

Funcionalidad del Nodo

Este nodo obtiene el tamaño actual de la VRAM disponible en bytes mediante la interfaz model_management de ComfyUI. Admite la ejecución de tres operaciones opcionales: vaciado suave de la caché, recolección forzada de basura y descarga de todos los modelos cargados. Estas operaciones están diseñadas para liberar recursos de VRAM ocupados. El nodo no procesa ni modifica ningún dato de entrada, sino que los pasa a través de sí mismo sin cambios.

Explicación de los Parámetros del Nodo - VRAM Debug

Entradas (Inputs)

Nombre del Parámetro	Tipo de Dato	Obligatorio	Valor por Defecto	Rango/Opciones	Descripción
`any_input`	IO	No	-	-	any_input (parámetro de entrada): Una interfaz de entrada genérica que puede recibir datos de cualquier tipo. El nodo no los procesa, solo se utiliza como una de las señales para activar la ejecución del nodo.
`image_pass`	IMAGE	No	-	-	image_pass (parámetro de salida): Devuelve los datos de imagen de entrada `image_pass` sin cambios.
`model_pass`	MODEL	No	-	-	model_pass (parámetro de salida): Devuelve los datos del modelo de entrada `model_pass` sin cambios.

Parámetros de Control (Parameters)

Nombre del Parámetro	Tipo de Dato	Obligatorio	Valor por Defecto	Rango/Opciones	Descripción
`empty_cache`	BOOLEAN	Sí	True	-	empty_cache (parámetro de entrada): Interruptor booleano que determina si se ejecuta la operación de vaciado suave de la caché para intentar liberar VRAM. Habilitado por defecto.
`gc_collect`	BOOLEAN	Sí	True	-	gc_collect (parámetro de entrada): Interruptor booleano que determina si se fuerza la recolección de basura (garbage collection) para liberar memoria. Habilitado por defecto.
`unload_all_models`	BOOLEAN	Sí	False	-	unload_all_models (parámetro de entrada): Interruptor booleano que determina si se descargan todos los modelos de la memoria de ComfyUI. Deshabilitado por defecto.

Salidas (Outputs)

Nombre del Parámetro	Tipo de Dato	Descripción
any_output	ANY	any_output (parámetro de salida): Devuelve los datos de entrada `any_input` sin cambios.
image_pass	IMAGE	image_pass (parámetro de salida): Devuelve los datos de imagen de entrada `image_pass` sin cambios.
model_pass	MODEL	model_pass (parámetro de salida): Devuelve los datos del modelo de entrada `model_pass` sin cambios.
freemem_before	INT	freemem_before (parámetro de salida): Número de bytes de VRAM libre en el sistema antes de ejecutar cualquier operación de limpieza.
freemem_after	INT	freemem_after (parámetro de salida): Número de bytes de VRAM libre en el sistema después de ejecutar las operaciones de limpieza especificadas.

Casos de Uso

En flujos de trabajo complejos, al generar múltiples imágenes de alta resolución de forma consecutiva o al cambiar entre diferentes modelos grandes, la VRAM puede agotarse gradualmente. En estos casos, se puede insertar este nodo en puntos clave del flujo de trabajo (por ejemplo, entre dos nodos que consumen muchos recursos). Al activar su funcionalidad de gestión de VRAM, se libera espacio, asegurando que los nodos posteriores puedan ejecutarse sin problemas. Además, se puede monitorizar la efectividad de la liberación mediante los valores enteros de salida.

Consideraciones

Este nodo está principalmente destinado a la depuración y monitorización de recursos. La operación de descarga de modelos que ejecuta (cuando está habilitada) afectará a otros nodos del flujo de trabajo que dependan de los modelos previamente cargados. Por lo tanto, se recomienda colocarlo en una fase del flujo de trabajo donde no sea necesario utilizar inmediatamente los modelos cargados anteriormente, o como un paso de limpieza final antes de terminar el flujo de trabajo.

Enlace al Código Fuente del Nodo VRAM Debug

El nodo VRAM Debug pertenece al paquete de nodos ComfyUI-KJNodes.

VRAM Debug(VRAM_Debug)

VRAM Debug