genie/configs/magvit_n32_h8_d64_action_concat.json

{
    "num_layers": 32,
    "num_heads": 8,
    "use_actions": true,
    "d_model": 64,
    "T": 12,
    "S": 256,
    "image_vocab_size": 262144,
    "use_mup": false,
    "action_network": "concat+modulate",
    "num_factored_vocabs": 2,
    "qkv_bias": false,
    "proj_bias": true,
    "attn_drop": 0.0,
    "qk_norm": false,
    "mlp_ratio": 4.0,
    "mlp_drop": 0.0,
    "mlp_bias": true
}