Target-Driven-Distillation

让扩散模型蒸馏更快更稳,生成质量不掉线

Target-Driven-Distillation 是一个面向扩散模型的蒸馏加速框架,核心思想是引入目标时间步选择与解耦引导,提升一致性蒸馏(Consistency Distillation)的训练效率与生成质量。它解决的是扩散模型采样速度慢、蒸馏过程收敛不稳定、引导信号耦合导致质量下降的问题。项目基于 PyTorch 实现,兼容 Stable Diffusion、SDXL、FLUX、LCM-LoRA 等主流架构,支持图像与视频生成场景。其核心能力包括:动态选择最优目标时间步以加速蒸馏收敛,将引导信号解耦为独立分支以减少干扰,并提供与现有 LoRA 蒸馏流程的无缝集成。代码结构清晰,适合研究蒸馏算法或工程化部署加速模型。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 112
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类图像生成
开发团队FireRedTeam
所属国家
定价模式free
价格说明开源项目,代码公开,可自行部署使用,无收费计划。
访问状态
是否开源是
开源协议
主要语言Python
技术栈/模型consistency-models,diffusion-models,flux,lcm-lora,lora,sdxl-lightning,stable-diffusion,stable-video-diffusion
GitHub 星标★ 112
30天Star增速
HF 下载量
上线时间2024-08-21 00:00:00
最近更新2026-08-28 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数3

使用教程

核心亮点

  • 提出目标时间步选择策略,相比固定步长蒸馏收敛速度明显提升
  • 解耦引导信号,减少分类器引导与蒸馏目标的相互干扰,生成细节更保真
  • 兼容主流扩散生态(SD/SDXL/FLUX),可直接复用现有 LoRA 训练流程

不足之处

  • 项目处于早期,文档仅覆盖基础用法,高级调参说明不足
  • 未提供完整训练好的蒸馏权重,需用户自行训练验证效果

适用场景

  • 将 SD/SDXL 模型蒸馏为少步采样版本,用于实时图像生成应用
  • 在视频扩散模型(如 Stable Video Diffusion)上加速采样,提升编辑效率
  • 研究一致性蒸馏算法,对比不同时间步选择策略对生成质量的影响

替代项目

Consistency Models、LCM-LoRA、SDXL-Lightning

项目介绍

Target-Driven-Distillation 是模型训练领域的开源项目,由 FireRedTeam 开发,2024 年首次发布。

它收录于模型训练分类,该分类目前共有 518 个项目,GitHub 星标数为 112。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-08-28。代码公开,可自行部署使用,无收费计划。

它主要面向的使用场景是:将SD/SDXL模型蒸馏为少步采样版本,用于实时图像生成应用。同类可对比的替代方案包括 Consistency Models、LCM-LoRA、SDXL-Lightning。

上一篇:TFMQ-DM

下一篇:safetensors_util

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13