File size: 4,299 Bytes
02fcdb4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | ---
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">
[-B31B1B?style=flat&logo=arXiv)](https://arxiv.org/abs/2401.01456)
[](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)
[-B31B1B?style=flat&logo=arXiv)](https://arxiv.org/abs/2504.06895)
[](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
@article{2024arXiv240101456Y,
author = {{Yan}, Dingkun and {Yuan}, Liang and {Wu}, Erwin and {Nishioka}, Yuma and {Fujishiro}, Issei and {Saito}, Suguru},
title = "{ColorizeDiffusion: Adjustable Sketch Colorization with Reference Image and Text}",
journal = {arXiv e-prints},
year = {2024},
doi = {10.48550/arXiv.2401.01456},
}
@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{yan2025colorizediffusionv2enhancingreferencebased,
title={ColorizeDiffusion v2: Enhancing Reference-based Sketch Colorization Through Separating Utilities},
author={Dingkun Yan and Xinrui Wang and Yusuke Iwasawa and Yutaka Matsuo and Suguru Saito and Jiaxian Guo},
year={2025},
journal = {arXiv e-prints},
doi = {10.48550/arXiv.2504.06895},
}
```
|