Jayashree Sridhar commited on
Commit
068c1e8
·
1 Parent(s): bc546a5

Added typegpt2 model in knowledgebase

Browse files
Files changed (1) hide show
  1. 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.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")),
 
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")),