data-archetype commited on
Commit
e816c03
Β·
verified Β·
1 Parent(s): 58b87c6

Fix model card: citation title, line break, add results viewer link

Browse files
Files changed (1) hide show
  1. README.md +4 -2
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 a skip-concat decoder with optional PDG sharpening.
 
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 with Skip-Concat Decoder},
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,