cde-small-v1 / config.json
mrloh's picture
Update config.json
8786de8 verified
{
"_name_or_path": "jxm/cde-small-v1",
"architecture": "transductive",
"architectures": [
"DatasetTransformer"
],
"attn_implementation": null,
"auto_map": {
"AutoConfig": "jxm/cde-small-v1--misc.ContextualModelConfig",
"AutoModel": "jxm/cde-small-v1--model.DatasetTransformer"
},
"biencoder_pooling_strategy": "mean",
"cache_dir": null,
"config_name": null,
"disable_dropout": true,
"disable_transductive_rotary_embedding": true,
"document_prompt": "search_document: ",
"embedder": "nomic-ai/nomic-bert-2048",
"embedder_rerank": "sentence-transformers/gtr-t5-base",
"embedding_output_dim": null,
"embedding_size": 768,
"limit_layers": null,
"limit_layers_first_stage": null,
"logit_scale": 50.0,
"max_seq_length": 512,
"model_revision": "main",
"query_prompt": "search_query: ",
"tokenizer_name": null,
"torch_dtype": "float32",
"transductive_corpus_size": 512,
"transductive_sequence_dropout_prob": 0.0,
"transductive_tie_token_embeddings": false,
"transductive_tokens_per_document": 1,
"transformers_version": "4.45.2"
}