Rethinking Preference Alignment for Diffusion Models with Classifier-Free Guidance

Environment setup

uv sync

Dataset

Model Inference

Inference on Stable Diffusion XL in cPGD method

bash scripts/infer_sdxl_cpgd.sh

Inference on Stable Diffusion XL in PGD method

bash scripts/infer_sdxl_pgd.sh

Model Training

This code is mainly built upon Diffusion-DPO, Diffusers.

Training on Stable Diffusion XL in cPGD method

bash scripts/train_sdxl_cpgd.sh

Training on Stable Diffusion XL in PGD method

bash scripts/train_sdxl_pgd.sh