Update config.py
Browse files
config.py
CHANGED
@@ -1,18 +1,18 @@
|
|
1 |
-
|
2 |
-
class ConfigConstants:
|
3 |
-
# Constants related to datasets and models
|
4 |
-
DATA_SET_NAMES = ['covidqa'
|
5 |
-
EMBEDDING_MODEL_NAME = "sentence-transformers/paraphrase-MiniLM-L3-v2"
|
6 |
-
RE_RANKER_MODEL_NAME = 'cross-encoder/ms-marco-electra-base'
|
7 |
-
GENERATION_MODEL_NAME = 'mixtral-8x7b-32768'
|
8 |
-
VALIDATION_MODEL_NAME = 'llama3-70b-8192'
|
9 |
-
GENERATION_MODELS = ["llama3-8b-8192", "qwen-2.5-32b", "mixtral-8x7b-32768", "gemma2-9b-it" ]
|
10 |
-
VALIDATION_MODELS = ["llama3-70b-8192", "deepseek-r1-distill-llama-70b" ]
|
11 |
-
DEFAULT_CHUNK_SIZE = 1000
|
12 |
-
CHUNK_OVERLAP = 200
|
13 |
-
|
14 |
-
class AppConfig:
|
15 |
-
def __init__(self, vector_store, gen_llm, val_llm):
|
16 |
-
self.vector_store = vector_store
|
17 |
-
self.gen_llm = gen_llm
|
18 |
self.val_llm = val_llm
|
|
|
1 |
+
|
2 |
+
class ConfigConstants:
|
3 |
+
# Constants related to datasets and models
|
4 |
+
DATA_SET_NAMES = ['covidqa', 'cuad']#, 'techqa' 'delucionqa', 'emanual', 'expertqa', 'finqa', 'hagrid', 'hotpotqa', 'msmarco', 'pubmedqa', 'tatqa']
|
5 |
+
EMBEDDING_MODEL_NAME = "sentence-transformers/paraphrase-MiniLM-L3-v2"
|
6 |
+
RE_RANKER_MODEL_NAME = 'cross-encoder/ms-marco-electra-base'
|
7 |
+
GENERATION_MODEL_NAME = 'mixtral-8x7b-32768'
|
8 |
+
VALIDATION_MODEL_NAME = 'llama3-70b-8192'
|
9 |
+
GENERATION_MODELS = ["llama3-8b-8192", "qwen-2.5-32b", "mixtral-8x7b-32768", "gemma2-9b-it" ]
|
10 |
+
VALIDATION_MODELS = ["llama3-70b-8192", "deepseek-r1-distill-llama-70b" ]
|
11 |
+
DEFAULT_CHUNK_SIZE = 1000
|
12 |
+
CHUNK_OVERLAP = 200
|
13 |
+
|
14 |
+
class AppConfig:
|
15 |
+
def __init__(self, vector_store, gen_llm, val_llm):
|
16 |
+
self.vector_store = vector_store
|
17 |
+
self.gen_llm = gen_llm
|
18 |
self.val_llm = val_llm
|