config.json

{"input_dim": 768, "hidden_size": 768, "output_dim": 2}