stivenDR14 commited on
Commit
10b3645
·
1 Parent(s): 9000413

update models with jambda

Browse files
Files changed (1) hide show
  1. utils.py +4 -4
utils.py CHANGED
@@ -9,20 +9,20 @@ 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-3.5-MoE-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
- "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
16
  "IBM Granite3.1 dense / Ollama local": "ollama",
17
  "Open AI / GPT-4o-mini": "openai",
18
  }
19
  else:
20
  AI_MODELS = {
21
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
22
- "Huggingface / Microsoft Phi 4 mini Instruct": "microsoft/Phi-3.5-MoE-instruct",
23
  "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
24
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
25
- "Huggingface / SmolLM2 1.7B Instruct": "HuggingFaceTB/SmolLM2-1.7B-Instruct",
26
  "Open AI / GPT-4o-mini": "openai",
27
  }
28
 
 
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
+ "Huggingface / Jamba Mini 1.6": "ai21labs/AI21-Jamba-Mini-1.6",
16
  "IBM Granite3.1 dense / Ollama local": "ollama",
17
  "Open AI / GPT-4o-mini": "openai",
18
  }
19
  else:
20
  AI_MODELS = {
21
  "Huggingface / Mistral Nemo Instruct": "mistralai/Mistral-Nemo-Instruct-2407",
22
+ "Huggingface / Microsoft Phi 3.5 Mini Instruct": "microsoft/Phi-3.5-mini-instruct",
23
  "Huggingface / Google Gemma 2 9B Instruct": "google/gemma-2-9b-it",
24
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
25
+ "Huggingface / Jamba Mini 1.6": "ai21labs/AI21-Jamba-Mini-1.6",
26
  "Open AI / GPT-4o-mini": "openai",
27
  }
28