Skip to content
This repository was archived by the owner on Feb 22, 2026. It is now read-only.

README.md

Latest commit

 

History

History
32 lines (22 loc) · 855 Bytes

README.md

File metadata and controls

32 lines (22 loc) · 855 Bytes

Pollux: Unified World Model

Pollux is a unified world-model training and generation codebase. A collaboration between KAUST and Nucleus on world models.

Install

  • Setup

Please see Install.md for a full installation.

  • Train
torchrun --standalone --nnodes 1 --nproc-per-node 8 -m apps.main.train \
  config=apps/main/configs/train_bucket_256_latent_code.yaml
  • Generate
python -m apps.main.generate config=apps/main/configs/eval.yaml

Docs