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

update models

Browse files
Files changed (1) hide show
  1. utils.py +4 -4
utils.py CHANGED
@@ -9,8 +9,8 @@ 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 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
13
- "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
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 +18,8 @@ if ENVIRONMENT == "dev":
18
  else:
19
  AI_MODELS = {
20
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
21
- "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
22
- "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
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 / 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
  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
  }