triencoder / final_model /config.json
triet-bit's picture
Upload final_model/config.json with huggingface_hub
4fe0bd6 verified
Raw
History Blame Contribute Delete
645 Bytes
{
"architectures": [
"PromptAwareTriEncoderModel"
],
"chunk_aggregation": "attention",
"context_model_id": "keepitreal/vietnamese-sbert",
"dropout": 0.3,
"dtype": "float32",
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1",
"2": "LABEL_2"
},
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
},
"max_chunks": 5,
"model_type": "patriencoder",
"num_head": 8,
"pooling_method": "mean",
"prompt_model_id": "keepitreal/vietnamese-sbert",
"response_model_id": "keepitreal/vietnamese-sbert",
"transformers_version": "4.56.2",
"use_response_prompt_attention": true,
"use_sbert": true
}