stivenDR14 commited on
Commit
f281286
·
1 Parent(s): b2ec6b4

update models

Browse files
Files changed (1) hide show
  1. utils.py +2 -4
utils.py CHANGED
@@ -9,8 +9,7 @@ ENVIRONMENT = os.getenv("ENVIRONMENT")
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
12
- "Huggingface / Microsoft Phi 4 Mini Instruct": "microsoft/Phi-4-mini-instruct",
13
- "Huggingface / Google Gemma 2 9B": "google/gemma-2-9b",
14
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
15
  "IBM Granite3.1 dense / Ollama local": "ollama",
16
  "Open AI / GPT-4o-mini": "openai",
@@ -18,8 +17,7 @@ if ENVIRONMENT == "dev":
18
  else:
19
  AI_MODELS = {
20
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
21
- "Huggingface / Microsoft Phi 4 Mini Instruct": "microsoft/Phi-4-mini-instruct",
22
- "Huggingface / Google Gemma 2 9B": "google/gemma-2-9b",
23
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
24
  "Open AI / GPT-4o-mini": "openai",
25
  }
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
12
+ "Huggingface / IBM Granite 3.2 2B Instruct": "microsoft/ibm-granite/granite-3.2-2b-instruct",
 
13
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
14
  "IBM Granite3.1 dense / Ollama local": "ollama",
15
  "Open AI / GPT-4o-mini": "openai",
 
17
  else:
18
  AI_MODELS = {
19
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
20
+ "Huggingface / IBM Granite 3.2 2B Instruct": "microsoft/ibm-granite/granite-3.2-2b-instruct",
 
21
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
22
  "Open AI / GPT-4o-mini": "openai",
23
  }