miniDiffusion:一个精简的Stable Diffusion 3.5 PyTorch复现
2025-06-14
miniDiffusion项目是一个使用纯PyTorch语言,仅依赖少量库的Stable Diffusion 3.5模型的重新实现。其代码精简,旨在用于教育、实验和研究目的。项目包含VAE、DiT、训练和数据集脚本等核心组件,并提供了模型训练和推理脚本。用户需安装依赖项并下载预训练模型权重。该项目已开源,并遵循MIT许可证。
AI