| { | |
| "codename": "rt-p-reg", | |
| "name": "RT-P Regressor", | |
| "task_type": "reg", | |
| "checkpoint_file": "model.safetensors", | |
| "embedding_model": "all-MiniLM-L12-v2", | |
| "d_text": 384, | |
| "model": { | |
| "num_blocks": 12, | |
| "d_model": 512, | |
| "d_text": 384, | |
| "num_heads": 8, | |
| "d_ff": 2048, | |
| "materialize_attn_masks": true | |
| } | |
| } | |