Repository navigation
v1.5 — ColorizeDiffusion fine-tuned baselines (200ep and 7ep)
ColorizeDiffusion v1/v2 fine-tuned on the Danbooru2021 training split (illustrations) and on ImageNet (natural images), the baselines of the paper:
*_200ep: fine-tuned with the common 200-epoch protocol, used for the ColDiff (200ep) rows of Tables II/III and the natural-image tables.*_7ep: fine-tuned with the authors' original 7-epoch schedule, used for supplementary Table X and the natural-image tables (7ep).- Names:
coldiff_<v1|v2>_<illust|real>_<dot|scr>_<200ep|7ep>.safetensors;real= ImageNet.coldiff_v1_real_dot_200epis the last saved state of a run that completed its 200 epochs (saved aslatest/).
Each file is split into parts below 2 GiB: cat <name>.part-* > <name>. SHA-256 of the whole files: checkpoints/README.md.
Inference: the official ColorizeDiffusion code with the wrappers described in reproduce/paper_experiments/coldiff_finetune/.
License: these weights are adapted from the official ColorizeDiffusion weights (https://github.com/tellurion-kanata/colorizeDiffusion) and are distributed under CC BY-NC-SA 4.0 (non-commercial, share-alike, attribution), not under the repository's MIT license.