draft_probe_suite/probe/config.json

{"input_size": 4096, "output_size": 1, "intermediate_size": 1024}