Spaces:
Sleeping
Sleeping
tangxuemei
commited on
Update src/backend/model_operations.py
Browse files
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'
|
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
|