stivenDR14 commited on
Commit
ca7c295
·
1 Parent(s): ce74d46

update models to use

Browse files
Files changed (1) hide show
  1. utils.py +4 -4
utils.py CHANGED
@@ -8,16 +8,16 @@ ENVIRONMENT = os.getenv("ENVIRONMENT")
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
- "Huggingface / IBM granite granite 3.1 8b Instruct": "ibm-granite/granite-3.1-8b-instruct",
12
- "Huggingface / Mistral Small 24B Instruct": "mistralai/Mistral-Small-24B-Instruct-2501",
13
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
14
  "IBM Granite3.1 dense / Ollama local": "ollama",
15
  "Open AI / GPT-4o-mini": "openai",
16
  }
17
  else:
18
  AI_MODELS = {
19
- "Huggingface / IBM granite granite 3.1 8b Instruct": "ibm-granite/granite-3.1-8b-instruct",
20
- "Huggingface / Mistral Small 24B Instruct": "mistralai/Mistral-Small-24B-Instruct-2501",
21
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
22
  "Open AI / GPT-4o-mini": "openai",
23
  }
 
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
+ "Huggingface / IBM Granite 3.2 2B Instruct": "ibm-granite/granite-3.2-2b-instruct",
12
+ "Huggingface / Phi 4 mini Instruct": "microsoft/Phi-4-mini-instruct",
13
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
14
  "IBM Granite3.1 dense / Ollama local": "ollama",
15
  "Open AI / GPT-4o-mini": "openai",
16
  }
17
  else:
18
  AI_MODELS = {
19
+ "Huggingface / IBM Granite 3.2 2B Instruct": "ibm-granite/granite-3.2-2b-instruct",
20
+ "Huggingface / Phi 4 mini Instruct": "microsoft/Phi-4-mini-instruct",
21
  "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
22
  "Open AI / GPT-4o-mini": "openai",
23
  }