cyx96 commited on
Commit
25328d1
·
1 Parent(s): f10e39e

added phi4

Browse files
Files changed (1) hide show
  1. src/backend/model_operations.py +1 -1
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():