tangxuemei commited on
Commit
f81b2fe
·
verified ·
1 Parent(s): 22c286a

Update src/backend/model_operations.py

Browse files
Files changed (1) hide show
  1. src/backend/model_operations.py +1 -1
src/backend/model_operations.py CHANGED
@@ -281,7 +281,7 @@ class SummaryGenerator:
281
  def generate_summary(self, system_prompt: str, user_prompt: str):
282
  # Using Together AI API
283
  using_together_api = False
284
- together_ai_api_models = ['mixtral', 'dbrx', 'wizardlm', 'llama-3']
285
  for together_ai_api_model in together_ai_api_models:
286
  if together_ai_api_model in self.model_id.lower():
287
  using_together_api = True
 
281
  def generate_summary(self, system_prompt: str, user_prompt: str):
282
  # Using Together AI API
283
  using_together_api = False
284
+ together_ai_api_models = ['mixtral', 'dbrx', 'wizardlm']
285
  for together_ai_api_model in together_ai_api_models:
286
  if together_ai_api_model in self.model_id.lower():
287
  using_together_api = True