GenerativeModels for Face generation

Training on CelebA-HQ images to generate faces

• DDPM - implementation of original DDPM with minor things taken from IDDPM

• Result

• To train the model with specific config

CUDA_VISIBLE_DEVICES=0,1 torchrun \
    --standalone \
    --nproc_per_node=2 \
    -m ddpm.train_diffusion \
    --config ddpm/configs/config_2.yaml

Latent Diffusion

Under construction