Spaces:
Sleeping
Sleeping
added missing configs
Browse files- configs/auto_merging.yaml +8 -0
- configs/basic.yaml +7 -0
- configs/sentence_window.yaml +8 -0
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
|