leo-pasi commited on
Commit
eb2b59d
·
1 Parent(s): a7fd659

added missing configs

Browse files
configs/auto_merging.yaml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ source_doc: "Master_Thesis.pdf"
2
+ rag_mode: "auto-merging retrieval"
3
+ llm_openai_model: "gpt-4o-mini"
4
+ embed_model: "BAAI/bge-small-en-v1.5"
5
+ chunk_sizes: [2048, 512, 128]
6
+ similarity_top_k: 6
7
+ rerank_model: "cross-encoder/ms-marco-MiniLM-L-2-v2"
8
+ rerank_top_n: 2
configs/basic.yaml ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ source_doc: "Master_Thesis.pdf"
2
+ rag_mode: "classic retrieval"
3
+ llm_openai_model: "gpt-4o-mini"
4
+ embed_model: "BAAI/bge-small-en-v1.5"
5
+ similarity_top_k: 6
6
+ rerank_model: "cross-encoder/ms-marco-MiniLM-L-2-v2"
7
+ rerank_top_n: 2
configs/sentence_window.yaml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ source_doc: "Master_Thesis.pdf"
2
+ rag_mode: "sentence window retrieval"
3
+ llm_openai_model: "gpt-4o-mini"
4
+ embed_model: "BAAI/bge-small-en-v1.5"
5
+ sentence_window_size: 3
6
+ similarity_top_k: 6
7
+ rerank_model: "cross-encoder/ms-marco-MiniLM-L-2-v2"
8
+ rerank_top_n: 2