T-WLA-WM
The world model (world head).
One file, T-WLA-WM.pt: a PyTorch checkpoint {"model": state_dict} holding the world.* tensors, 32-bit weights.
huggingface-cli download anon-submission-0518/T-WLA-WM T-WLA-WM.pt --local-dir checkpoints
Place it at checkpoints/T-WLA-WM.pt in the code repository. Joint training starts the world head from it:
torchrun --nproc_per_node=4 train.py --config configs/train.json --init-world-from ../checkpoints/T-WLA-WM.pt
The final model is in anon-submission-0518/T-WLA-FD.