Description
Un nœud expérimental qui corrige le mécanisme d'auto-attention de WanVideo pour utiliser une implémentation sageattention personnalisée, réduisant ainsi l'utilisation maximale de la mémoire VRAM lors de l'inférence. Ce nœud remplace le mode d'attention du modèle.
Nécessite la dernière version de sageattention – assurez-vous qu'elle est installée, compatible et à jour avant d'utiliser ce nœud.
Entrées
| Nom | Type | Description |
|---|---|---|
model | MODEL | Le modèle WanVideo auquel le correctif d'attention Sage à mémoire efficace est appliqué. |
Sorties
| Type | Description |
|---|---|
MODEL | La même instance de modèle avec son mécanisme d'auto-attention corrigé pour utiliser sageattention. |
Remarques d'utilisation
- Ce nœud est expérimental – son comportement peut changer dans les futures versions.
- Il n'affecte que les calques d'auto-attention de WanVideo ; l'attention croisée ou d'autres composants restent inchangés.
- Le correctif est appliqué comme un wrapper de modèle ; la
sortieMODELpeut être utilisée directement dans les nœuds d'échantillonnage suivants. - Si
sageattentionn'est pas installé ou est obsolète, une erreur sera levée lors de l'exécution. - L'application de ce correctif peut altérer la qualité ou le comportement de la sortie – testez soigneusement avec votre flux de travail spécifique.
Commentaires
Connectez-vous avec GitHub pour rejoindre la discussion.