StableDiffusion-PyTorch

从零手写 Stable Diffusion，彻底搞懂扩散模型原理

StableDiffusion-PyTorch 是一个用 PyTorch 从零实现 Stable Diffusion 模型的教学型开源项目，涵盖文本编码器、U-Net 扩散模型、VAE 自编码器及采样调度等核心组件。它解决了开发者想深入理解 Stable Diffusion 内部原理但官方代码复杂难懂的问题，通过简洁清晰的代码结构，将扩散模型的前向加噪、反向去噪、交叉注意力等关键机制逐一拆解。项目提供完整的训练与推理流程，支持自定义数据集微调，并附带详细注释，适合作为学习扩散模型和图像生成的入门教材。其核心能力在于将论文算法转化为可运行的 PyTorch 代码，帮助用户从零搭建自己的生成模型，而非直接提供高性能生成效果。