shortcut-models:基于 JAX 的扩散模型加速项目

可用于加速扩散模型和流匹配模型的采样过程，通过单网络和单训练阶段实现单步或多步高质量样本生成，支持不同采样步预算，相比传统方法提升效率且简化训练复杂度。【此简介由AI生成】