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, MIT) |
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.