| --- |
| license: apache-2.0 |
| --- |
| |
| # UNet |
|
|
| A lightweight UNet with single-block levels and sliding window attention. |
|
|
| - Pixel-space model in CIELAB color space |
| - LAB input, RGB output |
| - Decompose the input images into their frequency-domain components |
| - [Docling](https://huggingface.co/ibm-granite/granite-docling-258M) as text encoder |
| - Token efficient visual text inputs |
| - Variable head in the attention modules across the layers |
|
|
| ## Retrospection |
|
|
| Reconstruction quality, from good to worst: |
| - U-Docling (this repo) |
| - U-DAE |
| - U-DAE-NLL |
| - EQ-SAE-CIELAB |
| - EQ-SAE-CIELAB-c8 |
| - VAE-f16-c4-kv |
| - VAE-f16-c4 |
| - VAE-f16-c8 |
|
|
| ## References |
|
|
| - 2411.17459 |
| - 2503.11576 |
| - 2510.17800 |
| - 2510.18279 |
|
|
