File size: 1,091 Bytes
7d67abf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | ---
library_name: pytorch
tags: ["robotics", "world-model", "video-generation", "causal-forcing", "oscar", "repa"]
---
# needed_checkpoints_robotics_repa
The three checkpoints needed to run the DMD + REPA stage (`release/stage_configs/3_dmd_merlin.yaml`, branch
`release/merlin-causal-forcing` of causal_control_cosmos_causalforcingpp).
| path | config key | what |
|---|---|---|
| `models/oscar_2b/oscar_2b_dcp2torch.pt` | `cosmos_ckpt` | OSCAR-2B backbone (real/fake score init) |
| `causal_cd_oscar_chunk3/t1000g/checkpoint_model_002000/model.pt` | `generator_ckpt` | causal CD generator, step 2000 (DMD init) |
| `OmniStream/` | `encoder_kwargs.align_encoder_path` | frozen REPA target encoder ([StreamFormer/OmniStream](https://huggingface.co/StreamFormer/OmniStream), MIT) |
```bash
hf download csusupergear/needed_checkpoints_robotics_repa --local-dir $HF_DIR
sha256sum -c $HF_DIR/SHA256SUMS
```
The layout matches `<HF_DIR>` in the release configs, so `release/tools/relocate_paths.py` fills `cosmos_ckpt` and
`generator_ckpt`; set `align_encoder_path` to `$HF_DIR/OmniStream`.
|