xtudbxk commited on
Commit
ccca9eb
·
verified ·
1 Parent(s): 7ea32c4

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +65 -0
README.md CHANGED
@@ -1,3 +1,68 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ library_name: pytorch
4
+ tags:
5
+ - image-generation
6
+ - diffusion
7
+ - class-conditional
8
+ - imagenet
9
+ - pytorch
10
  ---
11
+
12
+ # LDM-is-AE: Latent Diffusion is an Intrinsic Auto-Encoder
13
+
14
+ Released **inference checkpoints** for *LDM-is-AE* (NeurIPS 2026).
15
+ Code: https://github.com/PolyU-VCLab/LDMisAE
16
+
17
+ ## TL;DR
18
+
19
+ Latent diffusion is usually a two-stage pipeline -- train a VAE, then train a diffusion model in its
20
+ latent space. **LDM-is-AE removes the pipeline**: the DiT backbone is split into **DiT-E** / **DiT-D**
21
+ and the intermediate feature is supervised in the **image domain at every timestep**, so the backbone
22
+ itself becomes an intrinsic auto-encoder trained end-to-end in a single stage.
23
+
24
+ ## Checkpoints
25
+
26
+ | File | Resolution | Keys | Size |
27
+ | --- | --- | --- | --- |
28
+ | `LDMisAE.256.ckpt` | 256x256 | `model_ema1`, `args`, `epoch` | 3.90 GB |
29
+ | `LDMisAE.512.ckpt` | 512x512 | `model_ema1`, `args`, `epoch` | 3.93 GB |
30
+
31
+ Class-conditional ImageNet models. The files are **ema1-only** (inference); resuming training needs a
32
+ full checkpoint.
33
+
34
+ ## Results
35
+
36
+ | Setting | FID | IS |
37
+ | --- | --- | --- |
38
+ | ImageNet 256x256, class-conditional | **1.80** | **314** |
39
+ | ImageNet 512x512, class-conditional | **1.90** | **320** |
40
+ | Reconstruction (gFID) | **1.82** | -- |
41
+
42
+ ## Usage
43
+
44
+ ```bash
45
+ git clone https://github.com/PolyU-VCLab/LDMisAE.git && cd LDMisAE
46
+ pip install -r requirements.txt
47
+ huggingface-cli download xtudbxk/LDMisAE LDMisAE.256.ckpt --local-dir weights
48
+ CKPT=weights/LDMisAE.256.ckpt IMG_SIZE=256 CFG=2.25 NUM_IMAGES=50000 bash scripts/inference.sh
49
+ ```
50
+
51
+ Keep `--ema_mode ema1` (the default); `none` / `ema2` fail fast with a clear error. The LPIPS `vgg.pth`
52
+ must be reachable before sampling -- see the repository README (*Quick Start*).
53
+
54
+ ## Citation
55
+
56
+ ```bibtex
57
+ @inproceedings{ldm_is_ae_2026,
58
+ title = {LDM-is-AE: Latent Diffusion is an Intrinsic Auto-Encoder for End-to-End Image Generation},
59
+ author = {Zhang, Zhengqiang and Sun, Lingchen and Wu, Rongyuan and Yi, Qiaosi and
60
+ Kong, Xiangtao and Xiao, Chaodong and Zhang, Lei},
61
+ booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
62
+ year = {2026}
63
+ }
64
+ ```
65
+
66
+ ## License
67
+
68
+ Code is **Apache-2.0** (https://github.com/PolyU-VCLab/LDMisAE). Weights are released for research use.