modular-diffusion

像搭积木一样,快速搭建你的自定义扩散模型

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 12 🐛 待处理 12 👀 关注 7
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 292
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源MIT许可证,完全免费,可自行部署使用。

项目预览

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

项目数据

分类图像生成
开发团队cabralpinto
定价模式🏷️完全免费
价格说明开源MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型audio-generation,deep-learning,diffusion-models,image-generation,machine-learning,modular-design,python,pytorch,text-generation,transformer,u-net
GitHub 星标★ 292
上线时间2023-06-22 00:00:00
代码更新2025-06-17
仓库体积31.9 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

modular-diffusion 是一个基于 PyTorch 的模块化扩散模型设计与训练库。它旨在解决扩散模型研究中的代码复用和灵活定制问题,提供高度解耦的组件,如噪声调度器、采样器、UNet 骨干网络、条件嵌入等,用户可像搭积木一样组合出适合自己任务的扩散模型。该库支持图像生成、音频生成和文本生成等多种模态,内置常用训练循环和评估工具,降低从零实现扩散模型的门槛。其核心能力在于模块化设计,允许研究者快速替换或实验不同组件,而无需重写整个框架。适合学术研究、快速原型验证以及教学场景。

Star 增长趋势

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

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

核心亮点

  • 高度模块化设计,各组件可独立替换,方便实验对比
  • 支持图像、音频、文本多模态生成,适用面广
  • 基于 PyTorch,与主流生态无缝集成,上手快

不足之处

  • 早期项目,文档和示例尚不完善,学习曲线较陡
  • 社区规模小,遇到问题可参考的资源有限

适用场景

  • 学术研究中快速验证新的扩散模型架构
  • 工业界定制特定场景的生成模型(如音频、图像)
  • 教学演示扩散模型原理与组件交互

替代项目

Hugging Face Diffusers、OpenAI Guided Diffusion、Stable Diffusion (CompVis)

上一篇:Scaling-Diffusion-Transformers-muP

下一篇:BitVAE

同类项目推荐

TorchCP 开源

给深度学习预测加上可证明的可靠性边界,一行代码搞定不确定性量化。

A Python toolbox for conformal prediction research on deep learning models, using Py···

★ 482 2026-08-09
GEM 开源

微调时保住模型创造力,让生成不再千篇一律

Code for Paper (Preserving Diversity in Supervised Fine-tuning of Large Language Mod···

★ 62 2026-08-09
UniDiffusion 开源

一站式训练扩散模型,切换多种个性化方法无需改代码

A Diffusion training toolbox based on diffusers and existing SOTA methods, including···

★ 84 2026-08-09
lmms-engine 开源

一套引擎搞定多模态模型训练,快速上手,灵活扩展。

A simple, unified multimodal models training engine. Lean, flexible, and built for h···

★ 833 2026-08-09
unsloth 开源

本地一键跑起最新大模型,调教它不花一分钱

Local UI to run and train LLMs and diffusion models. Supports GGUF, MLX, Qwen3.8, De···

★ 79231 2026-08-09
chronos-forecasting 开源

像用语言模型一样,零样本预测任意时间序列

Chronos: Pretrained Models for Time Series Forecasting

★ 6277 2026-08-09
Torch-Pruning 开源

一键剪枝任意 PyTorch 模型,自动处理复杂依赖,压缩提速不降准。

[CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models···

★ 3369 2026-08-09
selfcodealign 开源

让代码模型自己教自己,无需人工标注,自动提升生成质量。

[NeurIPS'24] SelfCodeAlign: Self-Alignment for Code Generation

★ 325 2026-08-09