LoRA Qwen-Image 2.1 6 étapes de Pruna : pack ComfyUI officiel

Exemples de la fiche officielle du modèle à 1024.

Lorsque PrunaAI a publié ces adaptateurs pour la première fois en septembre, ils arrivaient sous forme de fichiers diffusers/PEFT avec l'alpha stocké dans les métadonnées safetensors, que le chargeur LoRA de ComfyUI ne lit pas, et la fiche du modèle indiquait qu'il n'existait aucun paquet ComfyUI officiel. Ces deux affirmations changent désormais : les adaptateurs se présentent comme des fichiers ComfyUI natifs fonctionnant à l'échelle prévue, et les flux de travail sont ceux de l'auteur plutôt qu'une conversion communautaire.

Deux adaptateurs 6 étapes

La v0.2 conserve l'ensemble de la v0.1 (5 étapes et 8 étapes) et ajoute une paire d'étudiants 6 étapes entraînés différemment :

PrunaAI indique que l'adaptateur 6 étapes standard améliore la paire v0.1 dans presque toutes les catégories. Le détail important est que les deux calendriers ne sont pas interchangeables : les adaptateurs v0.2 ont été entraînés sur leur propre calendrier sigma et la fiche déconseille de réutiliser les valeurs de la v0.1.

Le suréchantillonnage du prompt intégré au LoRA

La recette officielle de Qwen-Image 2.1 exécute un amplificateur de prompt avant la génération. L'adaptateur 6-steps-upsampled absorbe cette étape : un prompt court passe donc directement à l'échantillonnage et le temps de bout en bout diminue du coût d'un appel distinct à l'amplificateur.

La mesure séparée, dans la fiche, de l'étape de suréchantillonnage que l'adaptateur v0.2 est censé remplacer.

La fiche reste prudente sur cette affirmation. Elle qualifie cela de première distillation supprimant le besoin d'une passe de suréchantillonnage de prompt séparée, puis précise que l'adaptateur peut ne pas reproduire exactement ce que ferait le véritable suréchantillonneur, d'autres travaux étant promis. Pruna a aussi mesuré le suréchantillonneur seul, avec les amplificateurs officiels Qwen-Image-2.1-PE-T2I et PE-I2I, et avertit que l'addition des deux médianes ne donne pas un chiffre de bout en bout.

Résultats d'édition de la fiche officielle, la tâche pour laquelle l'adaptateur 6 étapes standard est réglé.

L'exportation ComfyUI officielle

Le référentiel ComfyUI contient quatre adaptateurs bf16, un par variante, et six flux de travail : génération texte-image et édition d'image pour les 6 étapes v0.2, les 8 étapes v0.1 et les 5 étapes v0.1. Chaque flux de travail charge un adaptateur à une puissance de 1.0 via le nœud natif LoraLoaderModelOnly, échantillonne avec euler à CFG 1 et utilise un calendrier ManualSigmas. Aucun nœud personnalisé n'est nécessaire.

L'arborescence est celle dont disposent déjà les utilisateurs de ComfyUI pour Qwen-Image 2.1, avec le checkpoint de base de Comfy-Org/Qwen-Image-2.1 :

ComfyUI/
└── models/
    ├── diffusion_models/
    │   └── qwen_image_2.1_int8_convrot.safetensors
    ├── loras/
    │   └── p_q_image_2.1_6_steps_v0.2_comfyui_bf16.safetensors
    ├── text_encoders/
    │   └── qwen3vl_8b_int8_convrot.safetensors
    └── vae/
        └── qwen_image_2.1_vae_bf16.safetensors

La fiche précise d'utiliser le checkpoint de base, et non celui de Turbo, ainsi qu'un calendrier par adaptateur :

Le même calendrier couvre les deux variantes 6 étapes : dans les flux de travail v0.2, passer lora_name de 6-steps à 6-steps-upsampled constitue toute la différence entre elles dans ComfyUI.

Ce qui n'a pas changé

• La couverture d'entraînement reste à 1K. Les adaptateurs ont été entraînés en 1024 avec jusqu'à trois images de référence pour l'édition. Pruna recommande de commencer en 1024x1024 et considère la 2K et les références plus nombreuses comme hors couverture, où la qualité peut varier.
• Jusqu'à 6,3x plus rapide reste le chiffre phare, sans CFG et sans prompt négatif. Les graphiques v0.2 sont de nouvelles mesures et la fiche note qu'elles n'ont pas été appariées à ceux de la v0.1.
• Rang 64 sur 232 cibles linéaires, dont proj_out, entraîné avec alpha 128 et l'échelle 2x intégrée dans les matrices B exportées. C'est pourquoi la puissance doit rester à 1.0.
• La v0.1 est toujours disponible. Les adaptateurs 5 étapes et 8 étapes sont inchangés, si bien qu'un flux de travail existant utilisant la conversion communautaire peut migrer vers les fichiers officiels sans changer son calendrier.