mindscience.diffuser

mindscience.diffuser.ConditionDiffusionTransformer

以 Transformer 作为骨干网络的条件控制扩散模型。

mindscience.diffuser.DDIMPipeline

DDIM 生成流程管线。

mindscience.diffuser.DDIMScheduler

DDIMScheduler 实现了去噪扩散概率模型 DDPM 中介绍的去噪过程。

mindscience.diffuser.DDPMPipeline

DDPM 生成流程管线。

mindscience.diffuser.DDPMScheduler

DDPMScheduler 实现了去噪扩散概率模型DDPM中介绍的去噪过程。

mindscience.diffuser.DiffusionScheduler

扩散调度器基类。

mindscience.diffuser.DiffusionTrainer

扩散训练器基类。

mindscience.diffuser.DiffusionTransformer

基于 Transformer 主干结构的扩散模型实现。