Add config.json (model metadata; enables Hub download counting)
Browse files- 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 |
+
}
|