Hugging Face's logo

hugfaceguy0001
/
SmallDiT

Model card Files
xet
 Community
hugfaceguy0001 commited on
Commit
c025faa
·
verified ·
1 Parent(s): a7d3247

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -2,4 +2,9 @@
2
  license: bigscience-openrail-m
3
  datasets:
4
  - ILSVRC/imagenet-1k
5
- ---
 
 
 
 
 
 
2
  license: bigscience-openrail-m
3
  datasets:
4
  - ILSVRC/imagenet-1k
5
+ ---
6
+ # SmallDiT
7
+ 复现经典的DiT工作([Scalable Diffusion Models with Transformers](https://arxiv.org/abs/2212.09748))，训练数据为ImageNet.
8
+ 代码仓库: `https://github.com/lixiang90/ClassicalModels`
9
+ ## vae
10
+ `vae.pt`是用于图像压缩的vae模型，把(256,256,3)的图像压缩为(32,32,4)的latents.