gourisankar85 commited on
Commit
d0b15a7
·
verified ·
1 Parent(s): db7e2f6

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +17 -17
config.py CHANGED
@@ -1,18 +1,18 @@
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
 
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