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():
|