Jayashree Sridhar
commited on
Commit
·
bc546a5
1
Parent(s):
4102f9d
Modified typegpt2 model in config
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 |
-
mistral_model=os.getenv("MISTRAL_MODEL", ModelConfig.
|
| 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 |
+
mistral_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")),
|