| ---
|
| license: cc-by-nc-sa-4.0
|
| tags:
|
| - sketch-colorization
|
| - diffusion
|
| - stable-diffusion-xl
|
| - image-to-image
|
| - reference-based
|
| ---
|
|
|
| # ColorizeDiffusion XL - Model Weights
|
|
|
| <div align="center">
|
|
|
| [](https://openaccess.thecvf.com/content/WACV2025/html/Yan_ColorizeDiffusion_Improving_Reference-Based_Sketch_Colorization_with_Latent_Diffusion_Model_WACV_2025_paper.html)
|
| [](https://arxiv.org/abs/2502.19937)
|
| [](https://arxiv.org/abs/2603.05971)
|
| [](https://github.com/tellurion-kanata/ColorizeDiffusionXL)
|
| [](https://huggingface.co/spaces/tellurion/ColorizeDiffusion)
|
|
|
| </div>
|
|
|
| Model weights for [ColorizeDiffusion XL](https://github.com/tellurion-kanata/ColorizeDiffusionXL), an SDXL-based reference-based sketch colorization framework (1024px).
|
| For the base SD2.1 implementation (512/768px), refer to the [original repository](https://github.com/tellurion-kanata/colorizeDiffusion) and [weights](https://huggingface.co/tellurion/ColorizeDiffusion).
|
|
|
| ## Model Variants
|
|
|
| | Filename prefix | Config | Description |
|
| |:----------------|:-------|:------------|
|
| | `sdxl-*` | `configs/inference/sdxl.yaml` | Base SDXL colorizer with embedding guidance |
|
| | `xlv2-*` | `configs/inference/xlv2.yaml` | SDXL colorizer with ReferenceNet adapter for enhanced background/style |
|
|
|
| ## Usage
|
|
|
| ### Inference with UI
|
|
|
| ```bash
|
| git clone https://github.com/tellurion-kanata/ColorizeDiffusionXL.git
|
| cd ColorizeDiffusionXL
|
| conda env create -f environment.yml
|
| conda activate hf
|
| python -u app.py
|
| ```
|
|
|
| Select and load a checkpoint from the UI dropdown. Weights are downloaded automatically from this repository.
|
|
|
| ### Online Demo
|
|
|
| Try the model directly on [Hugging Face Spaces](https://huggingface.co/spaces/tellurion/ColorizeDiffusion).
|
|
|
| ### Programmatic Download
|
|
|
| ```python
|
| from huggingface_hub import hf_hub_download
|
|
|
| path = hf_hub_download(
|
| repo_id="tellurion/ColorizeDiffusionXL",
|
| filename="<checkpoint_name>.safetensors",
|
| )
|
| ```
|
|
|
| ## License
|
|
|
| These weights are released under [CC-BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/).
|
|
|
| ## Citation
|
|
|
| ```bibtex
|
| @InProceedings{Yan_2025_WACV,
|
| author = {Yan, Dingkun and Yuan, Liang and Wu, Erwin and Nishioka, Yuma and Fujishiro, Issei and Saito, Suguru},
|
| title = {ColorizeDiffusion: Improving Reference-Based Sketch Colorization with Latent Diffusion Model},
|
| booktitle = {Proceedings of the Winter Conference on Applications of Computer Vision (WACV)},
|
| year = {2025},
|
| pages = {5092-5102}
|
| }
|
|
|
| @article{2025arXiv250219937Y,
|
| author = {{Yan}, Dingkun and {Wang}, Xinrui and {Li}, Zhuoru and {Saito}, Suguru and {Iwasawa}, Yusuke and {Matsuo}, Yutaka and {Guo}, Jiaxian},
|
| title = "{Image Referenced Sketch Colorization Based on Animation Creation Workflow}",
|
| journal = {arXiv e-prints},
|
| year = {2025},
|
| doi = {10.48550/arXiv.2502.19937},
|
| }
|
|
|
| @article{2026arXiv260305971Y,
|
| author = {{Yan}, Dingkun and {Wang}, Xinrui and {Iwasawa}, Yusuke and {Matsuo}, Yutaka and {Saito}, Suguru and {Guo}, Jiaxian},
|
| title = "{ColorizeDiffusion XL: Enhancing Embedding Guidance for Character Colorization and Geometry Disentanglement}",
|
| journal = {arXiv e-prints},
|
| year = {2026},
|
| doi = {10.48550/arXiv.2603.05971},
|
| }
|
| ```
|
|
|
