checkpoint_config.json

{
  "model_config": {
    "gemma_model_name": "google/gemma-3-4b-it",
    "cosmos_model_name": "nvidia/Cosmos-Predict2-2B-Video2World",
    "enable_world": true,
    "load_full_cosmos_pipeline": true
  },
  "freeze_config": {
    "freeze_gemma_vision": true,
    "freeze_gemma_language": true,
    "freeze_cosmos_vae": true
  }
}