Update car_racing/world_model/config.json
Browse files
car_racing/world_model/config.json
CHANGED
|
@@ -4,7 +4,9 @@
|
|
| 4 |
"model_type": "world_model",
|
| 5 |
"components": {
|
| 6 |
"vision_tokenizer": "vision_tokenizer.pt",
|
| 7 |
-
"world_model": "world_model.pt"
|
|
|
|
|
|
|
| 8 |
},
|
| 9 |
"hidden_state_dim": 256,
|
| 10 |
"latent_dim": 32
|
|
|
|
| 4 |
"model_type": "world_model",
|
| 5 |
"components": {
|
| 6 |
"vision_tokenizer": "vision_tokenizer.pt",
|
| 7 |
+
"world_model": "world_model.pt",
|
| 8 |
+
"encoder": "encoder.pt",
|
| 9 |
+
"decoder": "decoder.pt"
|
| 10 |
},
|
| 11 |
"hidden_state_dim": 256,
|
| 12 |
"latent_dim": 32
|