Hugging Face's logo

hzzheng
/
MaskDiT

code
Model card Files
xet
Community
MaskDiT / scripts /prepare_latent256.sh
devzhk
Add model files
972a35a
raw
history blame contribute delete
177 Bytes
 # Encode ImageNet 256x256 into latent space
python3 extract_latent.py --resolution 256 --ckpt assets/vae/autoencoder_kl.pth --batch_size 64 --outdir ../data/imagenet256-latent