jolibrain / joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications
https://www.joligen.com
Other
238 stars 31 forks source link

Consistency models with JoliGEN for inpainting #592

Closed beniz closed 8 months ago

beniz commented 9 months ago

This PR adds support for Consistency Models (https://arxiv.org/abs/2303.01469 and https://arxiv.org/abs/2310.14189).

Usage:

python3 train.py --dataroot /data1/beniz/data/noglasses2glasses_ffhq --checkpoints_dir /home/beniz/checkpoints --name noglasses2glasses --config_json examples/example_cm_noglasses2glasses.json

Example: image

Note: some options (and thus doc & examples scripts have been modified) to accomodate the BaseDiffusion main class.