config.json

{
  "action_dim": 2,
  "architectures": [
    "SIADWorldModel"
  ],
  "dropout": 0.1,
  "dtype": "float32",
  "ema_decay": 0.996,
  "ema_warmup_steps": 1000,
  "encoder_blocks": 8,
  "encoder_heads": 16,
  "encoder_mlp_dim": 4096,
  "in_channels": 8,
  "latent_dim": 1024,
  "model_type": "siad-world-model",
  "transformers_version": "5.2.0",
  "transition_blocks": 12,
  "transition_heads": 16,
  "transition_mlp_dim": 4096,
  "use_decoder": true
}