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](../docs/training.md)**. Start there. | |
| This directory contains: | |
| - `toml/sft_config/` β TOML recipes consumed by `cosmos_framework.scripts.train --sft-toml=β¦`. One file per recipe. The TOML is validated against the pydantic schema at [`cosmos_framework/configs/toml_config/sft_config.py`](../cosmos_framework/configs/toml_config/sft_config.py) at load time. | |
| - `launch_sft_*.sh` β paired launch shells. Each declares `TOML_FILE` plus `: "${DATASET_PATH:=β¦}"` / `: "${BASE_CHECKPOINT_PATH:=β¦}"` defaults (full repo-relative paths, matching what [`docs/training.md`](../docs/training.md) shows) and sources [`_sft_launcher_common.sh`](./_sft_launcher_common.sh), which sets the `torchrun` flags and forwards into `cosmos_framework.scripts.train`. `export`ing those vars in your shell before launching wins over the defaults; otherwise just run the shell after Steps 1+2 of `docs/training.md`. | |
| - `inference.py`, `inference_pipeline.py` β runnable inference helpers; see [docs/inference.md](../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` | | |