Skip to content

Environment Variables

  • IMAGINAIRE_OUTPUT_ROOT=outputs/train: Output root for training DCP checkpoints and logs. Paired example launchers default to the repository's outputs/train/ (or OUTPUT_ROOT when set); direct CLI/torchrun calls default to /tmp/cosmos-framework-output/ when this variable is unset. To resume an existing run, explicitly select its original output root and project/group/name. Existing checkpoints are not moved automatically. We recommend at least 1 TB of free disk space.
  • COSMOS_SMOKE=1: Enable smoke test.
  • COSMOS_VERBOSE=1: Enable verbose console output.
  • WANDB_API_KEY: Required when the recipe TOML's job.wandb_mode = "online". Recipes default to wandb_mode = "disabled"; flip the TOML and export this key to log to Weights & Biases.

Hugging Face:

  • [HF_HOME]: Directory for checkpoints and datasets. We recommend at least 1 TB of free disk space.

UV:

  • [UV_CACHE_DIR]: Directory for dependencies.