mrloh commited on
Commit
afcffd3
·
verified ·
1 Parent(s): 9232053

Add new SentenceTransformer model

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "jxm/cde-small-v1",
3
  "architecture": "transductive",
4
  "architectures": [
5
  "DatasetTransformer"
@@ -14,14 +14,17 @@
14
  "config_name": null,
15
  "disable_dropout": true,
16
  "disable_transductive_rotary_embedding": true,
 
17
  "embedder": "nomic-ai/nomic-bert-2048",
18
  "embedder_rerank": "sentence-transformers/gtr-t5-base",
19
  "embedding_output_dim": null,
 
20
  "limit_layers": null,
21
  "limit_layers_first_stage": null,
22
  "logit_scale": 50.0,
23
  "max_seq_length": 512,
24
  "model_revision": "main",
 
25
  "tokenizer_name": null,
26
  "torch_dtype": "float32",
27
  "transductive_corpus_size": 512,
 
1
  {
2
+ "_name_or_path": "./temp/cde-small",
3
  "architecture": "transductive",
4
  "architectures": [
5
  "DatasetTransformer"
 
14
  "config_name": null,
15
  "disable_dropout": true,
16
  "disable_transductive_rotary_embedding": true,
17
+ "document_prompt": "search_document: ",
18
  "embedder": "nomic-ai/nomic-bert-2048",
19
  "embedder_rerank": "sentence-transformers/gtr-t5-base",
20
  "embedding_output_dim": null,
21
+ "embedding_size": 768,
22
  "limit_layers": null,
23
  "limit_layers_first_stage": null,
24
  "logit_scale": 50.0,
25
  "max_seq_length": 512,
26
  "model_revision": "main",
27
+ "query_prompt": "search_query: ",
28
  "tokenizer_name": null,
29
  "torch_dtype": "float32",
30
  "transductive_corpus_size": 512,