nicklashansen commited on
Commit
381d507
·
verified ·
1 Parent(s): 1bd7742

Add config.json (model metadata; enables Hub download counting)

Browse files
Files changed (1) hide show
  1. config.json +29 -0
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "MMBench2 world model",
3
+ "model_type": "dreamer4",
4
+ "description": "Dreamer-4-style generative world model for large-scale multi-task continuous control, trained on the MMBench2 dataset (210 tasks). Accompanies the paper 'Hallucination in World Models is Predictable and Preventable'.",
5
+ "paper": "https://www.nicklashansen.com/mmbench2",
6
+ "dataset": "https://huggingface.co/datasets/nicklashansen/mmbench2",
7
+ "total_params": 350000000,
8
+ "observation": "224x224 RGB",
9
+ "action_dim": 16,
10
+ "num_tasks": 210,
11
+ "components": {
12
+ "tokenizer": {
13
+ "params": 100000000,
14
+ "file": "tokenizer.pt",
15
+ "description": "causal video tokenizer (~50M encoder + ~50M decoder) with a 64-dim tanh-bounded latent"
16
+ },
17
+ "dynamics": {
18
+ "params": 250000000,
19
+ "file": "dynamics.pt",
20
+ "description": "block-causal flow-matching Transformer with reward and behavior-cloning heads"
21
+ }
22
+ },
23
+ "variants": {
24
+ "base": "pretrained world model",
25
+ "coverage_aware": "coverage-aware finetuned world model",
26
+ "combined": "finetuned with all targeted data-collection sources"
27
+ },
28
+ "license": "mit"
29
+ }