Official pre-trained model weights of the paper titled "LaCon: Late-Constraint Diffusion for Steerable Guided Image Synthesis".
Model Weights Structure
``
LaCon/
└── data-preprocessing
├── bdcn.pth <----- Model weights of BDCN edge detector (Place in
data-preprocessing/bdcn-edge-detection/checkpoints)
├── vgg16.pth <----- Preresquisites of BDCN edge detector (Place in
data-preprocessing/bdcn-edge-detection/checkpoints)
└── u2net.pth <----- Model weights of U2-Net saliency detector (Place in
data-preprocessing/u2net-saliency-detection/checkpoints`)
└── sd-checkpoints
├── sd-v1-4.ckpt <----- SD v1.4
├── kl-f8.ckpt <----- VAE model of SD v1.4
├── sd-celebahq.ckpt <----- SD CelebA-HQ
└── vq-f4.ckpt <----- VAE model of SD CelebA-HQ
├── sdv14edge.pth <----- Condition aligner based on SD v1.4, for edge conditions
├── sdv14color.pth <----- Condition aligner based on SD v1.4, for color conditions
├── sdv14mask.pth <----- Condition aligner based on SD v1.4, for mask conditions
├── sdcelebedge.pth <----- Condition aligner based on SD CelebA-HQ, for edge conditions
└── sdceleb_stroke.pth <----- Condition aligner based on SD CelebA-HQ, for color conditions
评论