Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
cyx96
commited on
Commit
·
25328d1
1
Parent(s):
f10e39e
added phi4
Browse files
src/backend/model_operations.py
CHANGED
|
@@ -164,7 +164,7 @@ class SummaryGenerator:
|
|
| 164 |
using_replicate_api = False
|
| 165 |
replicate_api_models = ['snowflake', 'llama-3.1-405b']
|
| 166 |
using_pipeline = False
|
| 167 |
-
pipeline_models = ['llama-3.1', 'phi-3-mini','falcon-7b', 'phi-3.5', 'mistral-nemo', 'llama-3.3']
|
| 168 |
|
| 169 |
for replicate_api_model in replicate_api_models:
|
| 170 |
if replicate_api_model in self.model_id.lower():
|
|
|
|
| 164 |
using_replicate_api = False
|
| 165 |
replicate_api_models = ['snowflake', 'llama-3.1-405b']
|
| 166 |
using_pipeline = False
|
| 167 |
+
pipeline_models = ['llama-3.1', 'phi-3-mini','falcon-7b', 'phi-3.5', 'mistral-nemo', 'llama-3.3', 'phi-4']
|
| 168 |
|
| 169 |
for replicate_api_model in replicate_api_models:
|
| 170 |
if replicate_api_model in self.model_id.lower():
|