DMLP 1.0.3

Diffusion Model Learning Package

DMLP

DMLP is a python library for training diffusion model

• Website: https://yunhaoli12138.github.io/DMLP/
• Documentation: https://yunhaoli12138.github.io/DMLP/docs/
• Mailing list: [e-mail] , [e-mail]
• Source code: https://github.com/YunhaoLi12138/DMLP
• Contributing: Yunhao Li, Jieqi Liu
• Bug reports: https://github.com/YunhaoLi12138/DMLP/issues

It provides:

• APIs for constructing and training/fine-tuning text diffusion model
• Abstract classes for developing models in text diffusion

Installation

pip install DMLP

Tutorial

The detailed tutorial can be found on website: https://yunhaoli12138.github.io/DMLP/tutorial/

Reference

Codes and model architecture referred from: https://github.com/guangyliu/AE_diffusion.git