Analytic Diffusion Studio — 项目总览
Analytic Diffusion Studio — 项目总览
smalldiffusion 是一个轻量级扩散模型库，用不到 100 行核心代码实现了扩散模型的训练与采样。 本文档对项目进行全面技术解读，从整体架构到每个函数的实现细节。
文档结构
| 文件 | 内容 |
|---|---|
| 01_overview.md | 项目总览：架构设计、模块关系、扩散模型数学基础 |
| 02_diffusion.md | 核心模块：噪声调度、训练循环、采样算法 (diffusion.py) |
| 03_data.md | 数据模块：数据集工具、玩具数据集 (data.py) |
| 04_model_base.md | 模型基础：ModelMixin、预测模式修饰器、注意力机制、嵌入层 (model.py) |
| 05_model_dit.md | Diffusion Transformer 模型 (model_dit.py) |
| 06_model_unet.md | U-Net 模型 (model_unet.py) |
| 07_examples.md | 实战示例：从玩具模型到 Stable Diffusion |