LoRA de distillation Krea 2 Turbo en 4 étapes : réduction de 8 à 4

ComfyUI Wikinews

Un LoRA de distillation communautaire réduit Krea 2 Turbo de 8 étapes à 4 avec une texture au niveau du professeur. Poids LoRA standards, sans nœud personnalisé.

Krea2-Turbo-Distill-4step-LoRA (Hugging Face) par lvladikov est un LoRA de distillation par étape qui réduit le nombre d'étapes utilisables minimum de Krea 2 Turbo de 8 à 4, soit environ 1,6x plus rapide bout en bout (54,5 s contre 88,7 s à 1024x1024) avec une qualité de texture égale au professeur en 8 étapes.

Pourquoi un LoRA, et pourquoi c'était difficile

La distillation par étape exige généralement des changements concentrés dans la voie du pas de temps : une analyse du delta Raw-to-Turbo de Krea montre que time_embed.linear_2 change environ 5,5x plus que tout linéaire de bloc. Un LoRA qui figerait la voie du pas de temps retiendrait exactement les poids dont la tâche a le plus besoin, donc ce LoRA entraîne également la voie du pas de temps, tout en restant un LoRA standard de rang-64.

Le checkpoint actuel (chk00042000) ajoute deux améliorations par rapport aux versions antérieures :

  • Texture à l'égalité du professeur : l'énergie des détails fins est de 1,00x celle du professeur en 8 étapes à 1280x1280 et de 1,02x à 1440x1440, appariée bande pour bande sur le spectre de fréquence (le checkpoint précédent mesurait 0,88x et 0,82x).
  • Critique sensible aux prompts : l'entraînement évalue les images par rapport à leurs prompts, donc l'adhérence est directement renforcée plutôt qu'héritée.
8 steps4 steps plain4 steps + LoRA
Turbo 8 étapesTurbo 4 étapes : les détails souffrent4 étapes avec le LoRA : texture restaurée

Utilisation dans ComfyUI

Le dépôt livre des fichiers suffixés _comfyui à côté des poids diffusers :

  • krea2_turbo_4step_rank_64_lora_latest_comfyui.safetensors pour une utilisation normale (un pointeur tournant vers le checkpoint accepté le plus récent)
  • krea2_turbo_4step_rank_64_lora_chk00042000_comfyui.safetensors pour épingler ce checkpoint exact

Chargez-le au-dessus de Krea 2 Turbo, exécutez 4 étapes au lieu de 8, et gardez le guidage à 0,0. C'est un LoRA standard prêt à l'emploi : aucun nœud personnalisé, aucun sampler patché, aucun code. Douze résolutions entraînées de 512x512 à 1440x1440 sont publiées avec des balayages par résolution dans le dépôt.

Les gains atteignent même le mode aperçu 2 étapes hors spécification, où les fréquences médianes creuses de l'ancien checkpoint sont visiblement comblées.

Liens

Commentaires

Connectez-vous avec GitHub pour rejoindre la discussion.

Chargement des commentaires…
LoRA de distillation Krea 2 Turbo en 4 étapes : réduction de 8 à 4 | ComfyUI Wiki