1. Installation de Sage Attention 2.2 sur ComfyUI Portable [0]

thumbnail Ce guide explique comment installer et utiliser Sage Attention 2.2 pour accélérer les générations sur ComfyUI, en se concentrant d'abord sur la version portable.

1.1. Vérification des versions Python et Torch [4]

  1. Identifier les versions de Python et Torch utilisées par ComfyUI portable est la première étape avant de télécharger le package d'installation de Sage Attention. [5]
    1. Ouvrez le dossier de ComfyUI portable, puis le dossier "Python embedded". [7]
    2. Faites un clic droit dans une zone vide et sélectionnez "Ouvrir dans le terminal" pour ouvrir une invite de commande. [9]
    3. Pour connaître la version de Python, tapez la commande ./python.exe -V et appuyez sur Entrée. La version affichée est 3.12.9. [12]
    4. Pour connaître la version de Torch, tapez la commande ./python.exe -m pip show torch et appuyez sur Entrée. Cette commande affiche également la version de CUDA. [15]

1.2. Téléchargement du package Sage Attention [19]

  1. Sélectionner le bon package Sage Attention est crucial et dépend des versions de Python, Torch et CUDA identifiées. [18]
    1. Le lien de téléchargement se trouve dans la description de la vidéo. [21]
    2. Sur la page de téléchargement, recherchez le package Sage Attention approprié. [25]
    3. Le nom du fichier indique les versions compatibles : "CU 128" pour CUDA 12.8, "Torch 2.7.1" pour Torch 2.7.1, et "CP 312" pour Python 3.12. "Win AMD 64" spécifie la compatibilité avec Windows 64 bits. [28]
    4. Cliquez sur le fichier souhaité pour le télécharger et enregistrez-le. [33]

1.3. Installation de Sage Attention [36]

  1. Installer les prérequis et Sage Attention se fait via le terminal dans le dossier "Python embedded" de ComfyUI portable. [37]
    1. Installez d'abord le package Triton Windows en tapant la commande /python.exe -m pip install Triton-Windows et en appuyant sur Entrée. [41]
    2. Une fois Triton installé, installez Sage Attention. Copiez le chemin complet du fichier de téléchargement de Sage Attention, collez-le dans le terminal après la commande /python.exe -m pip install , puis appuyez sur Entrée. [45]
    3. L'installation de Sage Attention est terminée une fois que le terminal indique que l'opération a réussi. [46]

1.4. Utilisation de Sage Attention [48]

  1. Deux méthodes permettent d'activer Sage Attention : l'ajout d'un flag au lanceur de ComfyUI ou l'utilisation d'un nœud personnalisé. [49]
    1. Méthode par flag :
      • Ajoutez le flag --use-sage-attention au fichier de lancement de ComfyUI (par exemple, run Nvidia GPU). [51]
      • Collez le flag après l'argument -Windows-standalone-build, puis enregistrez le fichier. [55]
      • Lancez ComfyUI. Si Sage Attention est activé, le terminal affichera "Using Sage Attention". [57]
      • Aucun nœud supplémentaire n'est nécessaire avec cette méthode. [59]
    2. Méthode par nœud personnalisé (ComfyUI KJ nodes) :
      • Installez le pack de nœuds personnalisés "ComfyUI KJ nodes" via le gestionnaire de nœuds de ComfyUI. [60]
      • Après l'installation, redémarrez ComfyUI. [73]
      • Dans le workflow, recherchez et ajoutez le nœud "patch sage attention KJ". [77]
      • Connectez la sortie du modèle de "load checkpoint" à l'entrée du modèle de "patch sage attention KJ". [78]
      • Connectez ensuite la sortie du modèle de "patch sage attention KJ" à l'entrée du modèle de "K sampler". [79]
      • Dans le nœud "patch sage attention KJ", changez le paramètre "Sage attention" sur "auto". [80]
      • Cette méthode fonctionne également avec d'autres nœuds comme "load diffusion model" et "Unet Patcher", en connectant la sortie du modèle à l'entrée du nœud "patch sage attention KJ". [81]

2. Installation de Sage Attention 2.2 sur ComfyUI Desktop [93]

thumbnail Ce guide détaille l'installation de Sage Attention 2.2 pour la version bureau de ComfyUI, en suivant des étapes similaires à celles de la version portable.

2.1. Vérification des versions Python et Torch [95]

  1. Déterminer les versions de Python, Torch et CUDA utilisées par ComfyUI desktop est essentiel pour choisir le bon package Sage Attention. [95]
    1. Lancez la version bureau de ComfyUI. [96]
    2. Cliquez sur "toggle bottom panel" en haut à droite, puis sur "terminal". [98]
    3. Pour la version de Python, tapez python --version et appuyez sur Entrée. La version est 3.12.8. [100]
    4. Pour les versions de Torch et CUDA, tapez pip show torch et appuyez sur Entrée. Les versions indiquées sont Torch 2.7.0 et CUDA 12.8. [103]
    5. Le package Sage Attention à télécharger doit être compatible avec Python 3.12, Torch 2.7.0 et CUDA 12.8. [106]

2.2. Téléchargement et Installation de Sage Attention [107]

  1. Télécharger et installer Sage Attention pour la version bureau suit un processus similaire à la version portable. [107]
    1. Il n'y a pas de version de Sage Attention spécifiquement disponible pour Torch 2.7.0. Par conséquent, le package pour Python 3.12, Torch 2.7.1 et CUDA 12.8 sera utilisé. [108]
    2. Avant d'installer Sage Attention, installez le package Triton Windows en tapant pip install Triton-Windows dans le terminal et en appuyant sur Entrée. [111]
    3. Une fois Triton installé, installez Sage Attention. Copiez le chemin du fichier d'installation de Sage Attention depuis le dossier de téléchargement et collez-le dans le terminal ComfyUI après la commande pip install , puis appuyez sur Entrée. [114]
    4. L'installation est terminée lorsque le terminal confirme la réussite. [118]

2.3. Activation de Sage Attention sur ComfyUI Desktop [119]

  1. Utiliser Sage Attention sur la version bureau se fait de la même manière qu'avec la version portable, en installant les "ComfyUI KJ nodes" et en utilisant le nœud "patch Sage Attention KJ". [119]
    1. Le processus d'installation et d'utilisation du nœud est identique à celui décrit pour la version portable. [119]