Spaces:
Running on L40S
Running on L40S
Cosmos3 SFT Examples
Runnable artifacts for Cosmos3 supervised fine-tuning. The end-to-end walkthrough β data preparation, base-checkpoint conversion, launch, outputs, export to safetensors, and evaluation β lives in docs/training.md. Start there.
This directory contains:
toml/sft_config/β TOML recipes consumed bycosmos_framework.scripts.train --sft-toml=β¦. One file per recipe. The TOML is validated against the pydantic schema atcosmos_framework/configs/toml_config/sft_config.pyat load time.launch_sft_*.shβ paired launch shells. Each declaresTOML_FILEplus: "${DATASET_PATH:=β¦}"/: "${BASE_CHECKPOINT_PATH:=β¦}"defaults (full repo-relative paths, matching whatdocs/training.mdshows) and sources_sft_launcher_common.sh, which sets thetorchrunflags and forwards intocosmos_framework.scripts.train.exporting those vars in your shell before launching wins over the defaults; otherwise just run the shell after Steps 1+2 ofdocs/training.md.inference.py,inference_pipeline.pyβ runnable inference helpers; see docs/inference.md.
Recipe β launch shell
| Recipe | Launch shell |
|---|---|
| Vision SFT (Cosmos3-Nano) | launch_sft_vision_nano.sh |
| Vision SFT LoRA (Cosmos3-Super) | launch_sft_vision_super.sh |
| Reasoner Alignment SFT | launch_sft_llava_ov.sh |
| Reasoner Alignment SFT (Cosmos3-Nano) | launch_sft_videophy2_nano.sh |