Rethink Checkpoints

This repository organizes checkpoints by model role, backbone, task, and variant.

Rethink-Model/
├── MODEL_INDEX.json
├── e5/
│   ├── E5-A/
│   │   ├── clipscore/
│   │   ├── direct/
│   │   ├── hpsv2.1/
│   │   └── pickscore/
│   └── E5-B/
│       ├── EC4/{direct,i2i,t2i}/
│       └── UR2B/{direct,edit-r2,t2i}/
├── students/
│   ├── B1-{TI,TIM}-{Dance,Diffusion,Flow}/
│   └── B1-ZI-{TI,TIM}-{Dance,Diffusion,Flow}/
└── teachers/
    ├── rl/{sd35,zimage}/<task>/
    └── sft/{sd35,zimage}/<task>/

Directory guide

teachers/sft/<backbone>/<task>/ contains supervised fine-tuning teacher checkpoints.
teachers/rl/<backbone>/<task>/ contains reinforcement learning teacher checkpoints.
sd35 refers to SD3.5-Medium; zimage refers to Z-Image.
students/ contains student checkpoints. B1 denotes the SD3.5-Medium backbone and B1-ZI denotes the Z-Image backbone. The directory names also identify the student family (TI or TIM) and variant (Dance, Diffusion, or Flow). Each student directory has a final/ checkpoint; some also include a time-matched/ checkpoint.
e5/ contains E5-A and E5-B checkpoint groups. The nested directory names identify the scorer or objective and, where applicable, the direct, t2i, i2i, or edit-r2 variant.
MODEL_INDEX.json maps checkpoint identifiers to their directories and backbones.