Jayashree Sridhar
commited on
Commit
·
068c1e8
1
Parent(s):
bc546a5
Added typegpt2 model in knowledgebase
Browse files- utils/config.py +1 -1
utils/config.py
CHANGED
|
@@ -131,7 +131,7 @@ class Config:
|
|
| 131 |
|
| 132 |
# Model configuration
|
| 133 |
self.models = ModelConfig(
|
| 134 |
-
|
| 135 |
embedding_model=os.getenv("EMBEDDING_MODEL", ModelConfig.embedding_model),
|
| 136 |
whisper_model=os.getenv("WHISPER_MODEL", ModelConfig.whisper_model),
|
| 137 |
temperature=float(os.getenv("TEMPERATURE", "0.7")),
|
|
|
|
| 131 |
|
| 132 |
# Model configuration
|
| 133 |
self.models = ModelConfig(
|
| 134 |
+
tinygpt2_model=os.getenv("MISTRAL_MODEL", ModelConfig.tinygpt2_model),
|
| 135 |
embedding_model=os.getenv("EMBEDDING_MODEL", ModelConfig.embedding_model),
|
| 136 |
whisper_model=os.getenv("WHISPER_MODEL", ModelConfig.whisper_model),
|
| 137 |
temperature=float(os.getenv("TEMPERATURE", "0.7")),
|