| # Rethink Checkpoints |
|
|
| This repository organizes checkpoints by model role, backbone, task, and variant. |
|
|
| ```text |
| 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. |
|
|
