Fix model card: citation title, line break, add results viewer link
Browse files
README.md
CHANGED
|
@@ -13,7 +13,8 @@ library_name: mdiffae
|
|
| 13 |
|
| 14 |
**mDiffAE v2** β **M**asked **Diff**usion **A**uto**E**ncoder v2.
|
| 15 |
A fast, single-GPU-trainable diffusion autoencoder with a **96-channel**
|
| 16 |
-
spatial bottleneck and
|
|
|
|
| 17 |
**This is the recommended version** β it offers substantially better
|
| 18 |
reconstruction than [v1](https://huggingface.co/data-archetype/mdiffae-v1)
|
| 19 |
(+1.7 dB mean PSNR) while maintaining the same or better convergence for
|
|
@@ -26,6 +27,7 @@ Bottleneck: **96 channels** at patch size 16
|
|
| 26 |
## Documentation
|
| 27 |
|
| 28 |
- [Technical Report](technical_report_mdiffae_v2.md) β architecture, training changes from v1, and results
|
|
|
|
| 29 |
- [mDiffAE v1](https://huggingface.co/data-archetype/mdiffae-v1) β previous version
|
| 30 |
- [iRDiffAE Technical Report](https://huggingface.co/data-archetype/irdiffae-v1/blob/main/technical_report.md) β full background on VP diffusion, DiCo blocks, patchify encoder, AdaLN
|
| 31 |
|
|
@@ -117,7 +119,7 @@ recon = model.decode(latents, height=H, width=W, inference_config=cfg)
|
|
| 117 |
|
| 118 |
```bibtex
|
| 119 |
@misc{mdiffae_v2,
|
| 120 |
-
title = {mDiffAE v2: A Fast Masked Diffusion Autoencoder
|
| 121 |
author = {data-archetype},
|
| 122 |
year = {2026},
|
| 123 |
month = mar,
|
|
|
|
| 13 |
|
| 14 |
**mDiffAE v2** β **M**asked **Diff**usion **A**uto**E**ncoder v2.
|
| 15 |
A fast, single-GPU-trainable diffusion autoencoder with a **96-channel**
|
| 16 |
+
spatial bottleneck and optional PDG sharpening.
|
| 17 |
+
|
| 18 |
**This is the recommended version** β it offers substantially better
|
| 19 |
reconstruction than [v1](https://huggingface.co/data-archetype/mdiffae-v1)
|
| 20 |
(+1.7 dB mean PSNR) while maintaining the same or better convergence for
|
|
|
|
| 27 |
## Documentation
|
| 28 |
|
| 29 |
- [Technical Report](technical_report_mdiffae_v2.md) β architecture, training changes from v1, and results
|
| 30 |
+
- [Results β interactive viewer](https://huggingface.co/spaces/data-archetype/mdiffae-v2-results) β full-resolution side-by-side comparison
|
| 31 |
- [mDiffAE v1](https://huggingface.co/data-archetype/mdiffae-v1) β previous version
|
| 32 |
- [iRDiffAE Technical Report](https://huggingface.co/data-archetype/irdiffae-v1/blob/main/technical_report.md) β full background on VP diffusion, DiCo blocks, patchify encoder, AdaLN
|
| 33 |
|
|
|
|
| 119 |
|
| 120 |
```bibtex
|
| 121 |
@misc{mdiffae_v2,
|
| 122 |
+
title = {mDiffAE v2: A Fast Masked Diffusion Autoencoder},
|
| 123 |
author = {data-archetype},
|
| 124 |
year = {2026},
|
| 125 |
month = mar,
|