Skip to content

Releases: MegEngine/MegDiffusion

DDPM pretrained model

04 Aug 03:35
Compare
Choose a tag to compare

Try using MegDiffuson to sample images with DDPM pretrained model. (It's still a demo script now.)

python3 -m megdiffusion.pipeline.ddpm.sample --config ./configs/ddpm/cifar10.yaml

Other configs are available here.

Thanks to @Asthestarsfalll for converting original Tensorflow checkpoints to MegEngine (from @pesser ‘s PyTorch pretrained diffusion models and pretrained models on Huggingface provided by Google ).

DDPM demo

07 Jul 09:40
Compare
Choose a tag to compare
v0.0.1

🔧 action: add python publish workflow