diffusion_distiller

把扩散模型蒸馏成少步采样,生成速度提升数倍

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 34 🐛 待处理 4 👀 关注 5
GitHub ↗
GitHub 星标 ★ 260
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,无付费版本。

项目预览

diffusion_distiller 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队Hramchenko
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型denoising-diffusion,image-generation,model-distillation,v-diffusion
GitHub 星标★ 260
上线时间2022-01-18 00:00:00
代码更新2022-05-31
仓库体积6.7 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

diffusion_distiller 是一个基于 PyTorch 的扩散模型蒸馏工具,实现了论文《Progressive Distillation for Fast Sampling of Diffusion Models》中的 v-diffusion 方法。它解决扩散模型采样速度慢、推理成本高的问题,通过渐进式蒸馏将多步采样模型压缩为少步甚至单步模型,同时保持生成质量。核心能力包括:支持加载预训练扩散模型(如 ImageNet 上的 v-diffusion 模型)、执行多轮蒸馏训练、提供采样加速评估脚本,并兼容 denoising-diffusion 生态。项目代码结构清晰,适合研究蒸馏算法或加速扩散模型落地。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 260

技术标签

核心亮点

  • 实现了官方渐进蒸馏算法,可直接复现论文结果
  • 支持从预训练 v-diffusion 模型开始蒸馏,降低训练成本
  • 代码简洁,依赖少,易于二次开发或集成到现有项目

不足之处

  • 文档/社区待观察
  • 仅支持 v-diffusion 一种蒸馏变体,未覆盖其他蒸馏策略

适用场景

  • 研究扩散模型加速蒸馏算法
  • 将高成本扩散模型压缩为实时可用的生成模型
  • 在资源受限设备上部署高质量图像生成

替代项目

DeepFloyd/consistency_models、google-research/consistency_models、lucidrains/denoising-diffusion-pytorch

上一篇:Matlab-GAN

下一篇:PyTorch-CycleGAN

同类项目推荐

CICD-for-Machine-Learning 开源

用GitHub Actions一键跑通模型训练到部署全流程

A beginner's project on automating the training, evaluation, versioning, and deploym···

★ 123 2026-08-09
spiral 开源

让AI在对抗博弈中练出深度推理

SPIRAL: Self-Play on Zero-Sum Games Incentivizes Reasoning via Multi-Agent Multi-Tur···

★ 211 2026-08-09
Qwen-VL-Series-Finetune 开源

用自有数据微调 Qwen-VL,快速得到专用视觉大模型

An open-source implementaion for fine-tuning Qwen-VL series by Alibaba Cloud.

★ 1982 2026-09-16
OntoTune 开源

用本体知识给大模型开小灶,让专业回答更靠谱

[Paper][WWW2025] OntoTune: Ontology-Driven Self-training for Aligning Large Language···

★ 51 2026-08-09
WaDi 开源

一步生成高质量图像,蒸馏时精准把控权重方向

[CVPR 2026] Official PyTorch implementation of WaDi: Weight Direction-aware Distilla···

★ 128 2026-08-09
ART 开源

让AI智能体在实际任务中边干边学,用GRPO强化训练,快速提升多步决策能力。

Agent Reinforcement Trainer: train multi-step agents for real-world tasks using GRPO···

★ 10849 2026-08-10
shortened-llm 开源

压缩LLM,推理提速,显存减半,生成依旧流畅。

Compressed LLMs for Efficient Text Generation [ICLR'24 Workshop]

★ 91 2026-08-09
kedro-mlflow 开源

在 Kedro 管道中一键启用 MLflow 实验追踪与模型版本管理

A kedro-plugin for integration of mlflow capabilities inside kedro projects (especia···

★ 237 2026-08-09
J-Wash 开源

可视化模型内部机制,手动微调对齐行为并导出结果

Jacobian-Brainwash : A manual alignment tool for large language models built on Anth···

★ 242 2026-08-09