{ "d_model": 768, "num_heads": 12, "d_kv": 64, "d_ff": 3072, "num_layers": 12, "vocab_size": 32128, "relative_attention_num_buckets": 32, "relative_attention_max_distance": 128 }