config.json

{
  "cross_message_passing": true,
  "hidden_dim": 128,
  "init_dim": 128,
  "num_heads": 8,
  "num_layers": 10,
  "output_dim": 128,
  "pe_dim": 128,
  "positional_encoding": true
}