VRAM Debug (VRAM_Debug) - ComfyUI-KJNodes

Le nœud VRAM Debug récupère la quantité actuelle de VRAM libre (en octets) en appelant l'interface model_management de ComfyUI.

VRAM Debug est un nœud utilitaire conçu pour surveiller et gérer la mémoire vidéo (VRAM). Sa fonction principale est de mesurer et de rapporter la capacité de VRAM disponible pendant l'exécution d'un flux de travail, tout en offrant des options pour nettoyer la VRAM et décharger les modèles, aidant ainsi les utilisateurs à optimiser l'utilisation des ressources.

Fonctionnalités du nœud

Ce nœud récupère la quantité actuelle de VRAM libre (en octets) en appelant l'interface model_management de ComfyUI. Il prend en charge trois actions optionnelles : vider doucement le cache, forcer la collecte des déchets (garbage collection) et décharger tous les modèles chargés. Ces actions visent à libérer les ressources VRAM occupées. Le nœud ne traite ni ne modifie aucune donnée d'entrée ; il les transmet simplement telles quelles.

Explication des paramètres du nœud - VRAM Debug

Entrées (Inputs)

Nom du paramètre	Type de données	Requis	Valeur par défaut	Plage/Valeurs possibles	Description
`any_input`	IO	Non	-	-	any_input (paramètre d'entrée) : Une interface d'entrée générique qui peut recevoir tout type de données. Le nœud ne les traite pas, elles servent uniquement de signal pour déclencher l'exécution du nœud.
`image_pass`	IMAGE	Non	-	-	image_pass (paramètre de sortie) : Renvoie les données d'image d'entrée `image_pass` sans modification.
`model_pass`	MODEL	Non	-	-	model_pass (paramètre de sortie) : Renvoie les données de modèle d'entrée `model_pass` sans modification.

Paramètres de contrôle (Parameters)

Nom du paramètre	Type de données	Requis	Valeur par défaut	Plage/Valeurs possibles	Description
`empty_cache`	BOOLÉEN	Oui	Vrai	-	empty_cache (paramètre d'entrée) : Interrupteur booléen déterminant s'il faut exécuter une opération de vidage doux du cache pour tenter de libérer de la VRAM. Activé par défaut.
`gc_collect`	BOOLÉEN	Oui	Vrai	-	gc_collect (paramètre d'entrée) : Interrupteur booléen déterminant s'il faut forcer une collecte des déchets (garbage collection) du système. Activé par défaut.
`unload_all_models`	BOOLÉEN	Oui	Faux	-	unload_all_models (paramètre d'entrée) : Interrupteur booléen déterminant s'il faut décharger tous les modèles présents en mémoire dans ComfyUI. Désactivé par défaut.

Sorties (Output)

Nom du paramètre	Type de données	Description
any_output	ANY	any_output (paramètre de sortie) : Renvoie les données d'entrée `any_input` sans modification.
image_pass	IMAGE	image_pass (paramètre de sortie) : Renvoie les données d'image d'entrée `image_pass` sans modification.
model_pass	MODEL	model_pass (paramètre de sortie) : Renvoie les données de modèle d'entrée `model_pass` sans modification.
freemem_before	ENTIER	freemem_before (paramètre de sortie) : Nombre d'octets de VRAM libre du système avant l'exécution de toute opération de nettoyage.
freemem_after	ENTIER	freemem_after (paramètre de sortie) : Nombre d'octets de VRAM libre du système après l'exécution des opérations de nettoyage spécifiées.

Cas d'utilisation

Dans des flux de travail complexes, lors de la génération continue de plusieurs images haute résolution ou après le changement de différents grands modèles, la VRAM peut s'épuiser progressivement. Dans ce cas, ce nœud peut être inséré à des points clés du flux de travail (par exemple, entre deux nœuds consommateurs de ressources). En déclenchant ses fonctions de gestion de la VRAM, il libère de l'espace, garantissant l'exécution réussie des nœuds suivants, tout en permettant de surveiller l'efficacité de la libération via les valeurs entières en sortie.

Notes importantes

Ce nœud est principalement destiné au débogage et à la surveillance des ressources. Son opération de déchargement de modèles (lorsqu'elle est activée) affecte les autres nœuds du flux de travail qui dépendent des modèles déjà chargés. Il est donc recommandé de le placer dans une phase du flux de travail où les modèles précédemment chargés ne sont pas immédiatement nécessaires, ou de l'utiliser comme étape de nettoyage avant la fin du flux de travail.

Lien vers le code source du nœud VRAM Debug

Le nœud VRAM Debug provient du pack de nœuds ComfyUI-KJNodes.

VRAM Debug(VRAM_Debug)

VRAM Debug