nicklashansen commited on
Commit
1bd7742
·
verified ·
1 Parent(s): 57cc7e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ tags:
27
 
28
  The world model follows the architecture and two-stage training recipe of [Dreamer 4](https://arxiv.org/abs/2509.24527), adapted for large-scale multi-task continuous control, and is trained on **MMBench2** — a 427-hour, 210-task dataset for visual world modeling (see the [dataset repository](https://huggingface.co/datasets/nicklashansen/mmbench2)). Each variant is a `(tokenizer.pt, dynamics.pt)` pair at 224×224 resolution:
29
 
30
- - **tokenizer** — a causal video tokenizer (≈50M-parameter encoder + ≈50M-parameter decoder, projecting to a 64-dim tanh-bounded latent).
31
- - **dynamics** — a ≈250M-parameter block-causal Transformer trained on the frozen tokenizer with a shortcut flow-matching objective, with reward prediction and behavior cloning heads.
32
 
33
  ## Variants
34
 
 
27
 
28
  The world model follows the architecture and two-stage training recipe of [Dreamer 4](https://arxiv.org/abs/2509.24527), adapted for large-scale multi-task continuous control, and is trained on **MMBench2** — a 427-hour, 210-task dataset for visual world modeling (see the [dataset repository](https://huggingface.co/datasets/nicklashansen/mmbench2)). Each variant is a `(tokenizer.pt, dynamics.pt)` pair at 224×224 resolution:
29
 
30
+ - **tokenizer** — a causal video tokenizer (50M-parameter encoder + 50M-parameter decoder, projecting to a 64-dim continuous latent).
31
+ - **dynamics** — a 250M-parameter block-causal Transformer trained on the frozen tokenizer with a shortcut flow-matching objective.
32
 
33
  ## Variants
34