File size: 1,364 Bytes
454813e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | # TBSM pretrained assets
This directory contains the pretrained generator backbones and representation
models referenced by the TBSM training configurations. Stable Diffusion VAE
assets are intentionally excluded.
## Naming
Generator checkpoints follow
`{source}_{architecture}_{dataset}_{resolution}_{space}_{role}.pt`.
Representation checkpoints retain the public model family, training dataset,
and input resolution in their names.
## Source mapping
| Release name | Original asset |
| --- | --- |
| `jit_b16_in1k_256_pixel_backbone.pt` | JiT-B/16 ImageNet-1K 256px backbone |
| `jit_b16_in1k_256_pixel_fd_backbone.pt` | JiT-B/16 FD-loss tuning initialization |
| `pixeldit_xl_in1k_256_pixel_ema_backbone.pt` | PixelDiT-XL ImageNet-1K 256px EMA backbone |
| `pixeldit_xl_in1k_512_pixel_ema_backbone.pt` | PixelDiT-XL ImageNet-1K 512px EMA backbone |
| `ucgm_dit_xl2_in1k_256_sdvae_backbone.pt` | UCGM DiT-XL/2 ImageNet-1K 256px SD-VAE backbone |
| `ucgm_dit_xl4_in1k_512_sdvae_backbone.pt` | UCGM DiT-XL/4 ImageNet-1K 512px SD-VAE backbone |
| `mae_vit_b16_in1k_224.pth` | MAE ViT-B/16 ImageNet-1K checkpoint |
| `torchvision_resnet18_in1k_224.pth` | TorchVision ResNet-18 ImageNet-1K checkpoint |
| `siglip2_vit_b16_webli_256/` | SigLIP2 ViT-B/16 WebLI 256px checkpoint and config |
Use `sha256sum -c SHA256SUMS` from this directory to verify the files.
|