pmdlt commited on
Commit
542df44
·
verified ·
1 Parent(s): c5016f4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -27,5 +27,14 @@
27
  "transformers_version": "4.51.3",
28
  "use_cache": true,
29
  "use_sliding_window": false,
30
- "vocab_size": 151936
 
 
 
 
 
 
 
 
 
31
  }
 
27
  "transformers_version": "4.51.3",
28
  "use_cache": true,
29
  "use_sliding_window": false,
30
+ "vocab_size": 151936,
31
+ "rag_config": {
32
+ "docs_name_or_path": "pmdlt/MNLP_M3_rag_dataset",
33
+ "embedding_model": "pmdlt/MNLP_M3_document_encoder",
34
+ "retriever_type": "bi-encoder",
35
+ "similarity_fn": "cosine",
36
+ "top_k": 20,
37
+ "chunk_size": 256,
38
+ "num_chunks": 100000,
39
+ }
40
  }