Skip to content

v1.5 — ColorizeDiffusion fine-tuned baselines (200ep and 7ep)

Choose a tag to compare

@MADONOKOUKI MADONOKOUKI released this 02 Oct 00:45
· 2 commits to main since this release

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_200ep is the last saved state of a run that completed its 200 epochs (saved as latest/).

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.