decision_maker / model_config.json
HrushikeshGangane's picture
Publish decision_maker inference bundle
c800ccd verified
Raw History Blame Contribute Delete
330 Bytes
{
"format": "decision_maker.v1",
"model_config": {
"encoder_name": "bert-large-uncased",
"decision_layers": 4,
"decision_heads": 16,
"decision_ffn_dim": 4096,
"dropout": 0.1,
"score_dim": 256
},
"max_length": 256,
"temperature": 2.01273250579834,
"weights": "model.safetensors"
}