{
    "metadata": {
        "total_size": 6490177552
    },
    "weight_map": {
        "draft_model.embed_tokens.weight": "model-00001-of-00002.safetensors",
        "draft_model.emotion_vector": "model-00001-of-00002.safetensors",
        "draft_model.lm_head.weight": "model-00002-of-00002.safetensors",
        "draft_model.memory_keys": "model-00001-of-00002.safetensors",
        "draft_model.memory_values": "model-00001-of-00002.safetensors",
        "draft_model.norm.bias": "model-00001-of-00002.safetensors",
        "draft_model.norm.weight": "model-00001-of-00002.safetensors",
        "draft_model.prediction_head.bias": "model-00002-of-00002.safetensors",
        "draft_model.prediction_head.weight": "model-00002-of-00002.safetensors",
        "draft_model.response_selector.bias": "model-00002-of-00002.safetensors",
        "draft_model.response_selector.weight": "model-00002-of-00002.safetensors",
        "draft_model.shared_layer.input_layernorm.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.shared_layer.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.simulation_head.bias": "model-00002-of-00002.safetensors",
        "draft_model.simulation_head.weight": "model-00002-of-00002.safetensors",
        "draft_model.stimuli_vector": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.input_layernorm.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
        "draft_model.unique_layer.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
        "embed_tokens.weight": "model-00001-of-00002.safetensors",
        "emotion_vector": "model-00001-of-00002.safetensors",
        "lm_head.weight": "model-00001-of-00002.safetensors",
        "memory_keys": "model-00001-of-00002.safetensors",
        "memory_values": "model-00001-of-00002.safetensors",
        "norm.bias": "model-00001-of-00002.safetensors",
        "norm.weight": "model-00001-of-00002.safetensors",
        "prediction_head.bias": "model-00001-of-00002.safetensors",
        "prediction_head.weight": "model-00001-of-00002.safetensors",
        "response_selector.bias": "model-00001-of-00002.safetensors",
        "response_selector.weight": "model-00001-of-00002.safetensors",
        "shared_layer.input_layernorm.weight": "model-00001-of-00002.safetensors",
        "shared_layer.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
        "shared_layer.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
        "shared_layer.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
        "simulation_head.bias": "model-00001-of-00002.safetensors",
        "simulation_head.weight": "model-00001-of-00002.safetensors",
        "stimuli_vector": "model-00001-of-00002.safetensors",
        "unique_layer.input_layernorm.weight": "model-00001-of-00002.safetensors",
        "unique_layer.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
        "unique_layer.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
        "unique_layer.self_attn.v_proj.weight": "model-00001-of-00002.safetensors"
    }
}