smalldiffusion 实战示例

本章解读项目提供的所有示例,从 2D 玩具模型到 Stable Diffusion 级别的预训练模型。

示例数据模型调度条件运行方式
toyexample.ipynbSwissroll 2DTimeInputMLPLogLinearJupyter
cond_tree_model.ipynbTreeDataset 2DConditionalMLPLogLinear类别标签 + CFGJupyter
fashion_mnist_dit.pyFashionMNIST 28×28DiTDDPMaccelerate launch
fashion_mnist_dit_cond.pyFashionMNIST 28×28DiT + CondEmbedderDDPM类别标签 + CFGaccelerate launch
fashion_mnist_unet.pyFashionMNIST 28×28Scaled(Unet)LogLinearaccelerate launch
cifar_unet.pyCIFAR-10 32×32Scaled(Unet)Sigmoid(训练)/LogLinear(采样)accelerate launch
diffusers_wrapper.py-ModelLatentDiffusionLDM文本Python 模块
stablediffusion.py-ModelLatentDiffusionLDM文本python