Superprompt

Le nœud Superprompt est basé sur le modèle roborovski/superprompt-v1, un modèle T5 affiné sur le jeu de données SuperPrompt.

Le nœud Superprompt est un outil d'amélioration de texte qui peut automatiquement développer des invites textuelles courtes et simples en versions plus détaillées et descriptives. Sa fonction principale est d'utiliser un modèle de langage T5 affiné pour fournir des descriptions d'entrée plus riches pour les tâches ultérieures de génération d'images à partir de texte.

Fonctionnalités du nœud

Ce nœud est basé sur le modèle roborovski/superprompt-v1, un modèle T5 affiné sur le jeu de données SuperPrompt. Il reçoit une invite de base et une invite d'instruction, les combine et les soumet au modèle. Le modèle génère ensuite une sortie textuelle plus détaillée d'une longueur spécifiée, en fonction de l'instruction. Lors de la première utilisation, le nœud téléchargera automatiquement les fichiers de modèle nécessaires depuis Hugging Face.

Description des paramètres du nœud - Superprompt

Paramètres de contrôle (Parameters)

Nom du paramètre	Type de données	Requis	Valeur par défaut	Plage de valeurs/Options	Description
`instruction_prompt`	STRING	Oui	Expand the following prompt to add more detail	-	Le texte d'instruction qui guide le modèle sur la manière d'étendre l'invite. L'instruction par défaut est "Expand the following prompt to add more detail".
`prompt`	STRING	Oui	-	-	L'invite textuelle originale qui doit être étendue et améliorée. Ce paramètre reçoit généralement une entrée textuelle provenant d'autres nœuds.
`max_new_tokens`	INT	Oui	128	1 - 4096 (pas: 1)	Contrôle la longueur maximale du nouveau texte généré par le modèle. La plage de valeurs est de 1 à 4096, avec un pas de 1.

Sortie (Output)

Nom du paramètre	Type de données	Description
STRING	STRING	L'invite textuelle détaillée après extension et amélioration.

Cas d'utilisation

Dans un flux de travail typique de génération d'images IA, vous pouvez saisir une invite initiale courte (par exemple, "un chat") dans ce nœud. Il produira une description plus détaillée (par exemple, "un chat roux et duveteux, paresseusement allongé sur un rebord de fenêtre au soleil"). Cette invite améliorée peut être directement connectée au port d'entrée de l'invite du modèle de génération d'image à partir de texte pour produire une image plus riche en détails et plus conforme aux attentes.

Notes importantes

Veuillez noter que le niveau de détail et la qualité du texte généré sont influencés par le paramètre max_new_tokens et par l'invite d'entrée elle-même. Une génération trop longue peut entraîner de la redondance ou un manque de cohérence. Ce nœud nécessite une connexion Internet pour télécharger le modèle et le processus de génération consomme certaines ressources de calcul.

Lien vers le code source du nœud Superprompt

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