FLUX.3 Klein turbo PDD v0.1

Это few-step LoRA для FLUX.3 klein base от NoiseToNova, обученная через Parallel Decoding Distillation (PDD). Генерирует за 8 или 4 шага вместо 50, CFG дистиллирован (teacher CFG 4), поэтому каждый шаг — один forward pass. Версия v0.1, rank 128.

💪 Ключевые особенности

• 👉 8 шагов — лучший баланс, 4 — быстрее; 2 шага дают быстрый превью композиции с низкой детализацией.
• 💪 Сохраняет box control и text rendering base-модели.
• 👀 CFG уже внутри LoRA: при запуске ставьте cfg=1.
• 🐾 Обучена на text-to-image в 1K и multi-aspect ratio; image-to-image работает, но качество ниже base.
• 💅 Для ComfyUI есть отдельный safetensors и workflow; нужна нода ComfyUI-Flux3-PDD.

👀 Пример

После загрузки base-модели и обёртки PDDWrapper из карточки запуск сводится к:

pdd = PDDWrapper(pipe, config, state_dict)
steps = 8  # или 4
sigmas = pdd.set_steps(steps)
pipe.guider = pipe.guider.new(guidance_scale=1)
image = pipe(prompt='...', num_inference_steps=steps, sigmas=sigmas, output='images')[0]
image.save(f'knitted_{steps}.png')

Подробнее

LoRA rank 128; в файле также банк выходных голов proj_out.weight, которые сливаются на каждом jump.
Обучение: text-to-image, 1K, multi-aspect ratio; дистилляция CFG с teacher CFG 4.
Ограничения: лица, особенно мелкие и дальние, остаются слабым местом; image-to-image работает, но качество ниже base.
Для ComfyUI нужны версия 0.42+ и нода ComfyUI-Flux3-PDD; в diffusers — main-ветка, peft и base-модель.