Model Save KJ(ModelSaveKJ)
D'un point de vue technique, ce nœud reçoit une entrée de type `MODEL` et obtient le dictionnaire des paramètres du modèle via la méthode `state_dict_for_saving`.
Model Save KJ
Le nœud Model Save KJ, d'un point de vue technique, reçoit une entrée de type MODEL et obtient le dictionnaire de paramètres du modèle via la méthode state_dict_for_saving.
Model Save KJ est un nœud utilisé pour sauvegarder les poids d'un modèle de diffusion après traitement ou fusion. Sa fonction principale est de persister sur le disque, au format .safetensors, un modèle présent dans le flux de travail, tout en permettant à l'utilisateur de personnaliser le chemin de sauvegarde et le préfixe des clés internes du modèle. Ce nœud ne produit aucune sortie de données, il exécute uniquement l'opération de sauvegarde.
Fonctionnalités du nœud
D'un point de vue technique, ce nœud reçoit une entrée de type MODEL et obtient le dictionnaire de paramètres du modèle via la méthode state_dict_for_saving. Son mécanisme central réside dans le fait qu'il permet à l'utilisateur de spécifier un model_key_prefix, utilisé pour remplacer le préfixe par défaut des clés dans le dictionnaire de paramètres du modèle (model.diffusion_model.), puis de sauvegarder le dictionnaire de paramètres traité sous forme de fichier .safetensors. Cela offre une flexibilité pour un chargement ultérieur dans d'autres frameworks ou outils prenant en charge différentes conventions de nommage de clés.
Explication des paramètres du nœud - Model Save KJ
Entrées (Inputs)
| Nom du paramètre | Type de données | Requis | Valeur par défaut | Plage de valeurs/Options | Description |
|---|---|---|---|---|---|
model | MODEL | Oui | - | - | Spécifie l'objet modèle à sauvegarder. Il s'agit de la source de données centrale pour l'opération du nœud. |
Paramètres de contrôle (Parameters)
| Nom du paramètre | Type de données | Requis | Valeur par défaut | Plage de valeurs/Options | Description |
|---|---|---|---|---|---|
filename_prefix | STRING | Oui | diffusion_models/ComfyUI | - | Définit le préfixe du chemin et le nom de base du fichier de sauvegarde. Par défaut, il est enregistré dans le chemin diffusion_models/ComfyUI du répertoire output, et un numéro de séquence est automatiquement ajouté au nom du fichier. |
model_key_prefix | STRING | Oui | model.diffusion_model. | - | Spécifie la nouvelle chaîne de caractères utilisée pour remplacer le préfixe original des clés dans le fichier de modèle sauvegardé. La valeur par défaut est model.diffusion_model.. Modifier cette valeur permet d'adapter le modèle aux exigences de chargement de différents frameworks. |
Cas d'utilisation
Un cas d'utilisation typique intervient après un flux de travail de fusion ou de fine-tuning de modèles. Par exemple, après avoir fusionné et ajusté un modèle de base avec plusieurs modèles LoRA, vous pouvez sauvegarder le modèle final généré via ce nœud. Ainsi, vous pourrez charger directement ce modèle personnalisé dans d'autres projets ou dans d'autres flux de travail ComfyUI, sans avoir à répéter l'intégralité du processus de fusion complexe.
Lien vers le code source du nœud Model Save KJ
Le nœud Model Save KJ provient du pack de nœuds ComfyUI-KJNodes.
Commentaires
Connectez-vous avec GitHub pour rejoindre la discussion.