FlashControl: One-Step Controllable Generator via Distillation-Friendly Single-Stream Teachers

flashcontrol:-one-step-controllable-generator-via-distillation-friendly-single-stream-teachers

FlashControl: One-Step Controllable Generator via Distillation-Friendly Single-Stream Teachers

[Submitted on 15 Oct 2026]

View PDF HTML (experimental)

Abstract:Diffusion models have become a leading approach in visual generative modeling, particularly effective in text-to-image and controllable generation. Their main drawback is slow inference as it requires many sampling steps. While recent distillation methods enable one-step text-to-image generation, extending these techniques to controllable generation is still challenging. Existing controllable models, such as ControlNet, rely on dual-branch design that make distillation difficult. In this work, we adopt a simpler strategy. Instead of adding extra conditional branches, we finetune a selected subset of layers in the original text-to-image diffusion model to directly enable controllable generation. This results in DeltaControl, a single-flow, multi-step diffusion model capable of supporting spatial-condition generation while matching the performance of dual-branch approaches. Building on this teacher model, we further apply step-distillation to obtain FlashControl, a one-step model capable of spatial controllable generation with a single forward pass.

Submission history

From: Ngan Nguyen [view email]
[v1] Thu, 15 Oct 2026 18:32:24 UTC (14,591 KB)