Skip to content

Releases: MegEngine/MegDiffusion

DDPM pretrained model

Choose a tag to compare

@0xMashiro 0xMashiro released this 04 Aug 03:35

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

Choose a tag to compare

@0xMashiro 0xMashiro released this 07 Jul 09:40
v0.0.1

🔧 action: add python publish workflow