HF hosted SiT Checkpoints

[willisma]

Hi all,

We did a small refactorization and now all SiT checkpoints will be hosted on Hugginface. You can access them via

hf download nyu-visionx/SiT-collections --local-dir pretrained_models

For running experiments (e.g., sampling), simply pass in args.ckpt pointing to the downloaded model path. The HF space containing both model checkpoints in 256x256 and 512x512, and we will gradually roll out pretrained checkpoints for smaller model sizes.

[willisma]

Following up on this, we will deprecate the dropbox downloading link. Please stick to using the hf space and open an issue if any problem is encountered.

[willisma]

We have released S, B, L checkpoints in hf space as well. These models are trained at 800k steps and in 256x256 resolution.