config.json

{
  "depth": 1,
  "dropout": 0.0,
  "embed_dim": 1024,
  "heads": 8,
  "input_dim": 512,
  "lambda_kl": 0.001,
  "masked_only": false,
  "mlp_dim": 2048,
  "num_cls_tokens": 2,
  "num_modalities": 6,
  "num_register_tokens": 4,
  "transformers_version": "5.2.0"
}