config.json

{
  "dim_A": 768,
  "dim_B": 768,
  "hidden_dim": 512
}