HybridStitch: Token-Adaptive Model Stitching for Efficient Diffusion Generation

Desen Sun, Jason Hon, Jintao Zhang +1

Iterative diffusion generation requires different amounts of model capacity at different token positions. Prior model combinations allocate capacity by denoising step or by draftin…