Feature Extraction
Transformers
Safetensors
cue
user-simulation
dialogue
persona
custom_code
cue / config.json
AnjaliRuban's picture
Point example_pool_id at handshake-ai-research/cue-example-pool
7612195 verified
Raw History Blame Contribute Delete
1.17 kB
{
"architectures": [
"CueModel"
],
"auto_map": {
"AutoConfig": "configuration_cue.CueConfig",
"AutoModel": "modeling_cue.CueModel"
},
"command_slot_decode": true,
"decoder": {
"bottleneck_dim": 1024,
"dtype": "bfloat16",
"freeze_lm": false,
"general_command_slots": 5,
"injection": "flamingo_gated_xattn_dense",
"insert_every": 4,
"model_name": "Qwen/Qwen3-0.6B-Base",
"num_heads": 8,
"num_persona_tokens": 16,
"slot_embedding_dim": 64,
"style_command_slots": 5,
"user_specific_command_slots": 5
},
"dual_decode": true,
"encoder": {
"bottleneck_dim": 1024,
"max_length": 256,
"max_turns": 64,
"passage_prefix": "search_document: ",
"query_prefix": "search_query: ",
"session_dim": 1024,
"session_heads": 8,
"session_layers": 4,
"system_model_name": "nomic-ai/modernbert-embed-base",
"user_model_name": "nomic-ai/modernbert-embed-base"
},
"example_pool_id": "handshake-ai-research/cue-example-pool",
"general_command_slots": 5,
"model_type": "cue",
"slot_embedding_dim": 64,
"style_command_slots": 5,
"user_specific_command_slots": 5
}