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
|