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

update mistral

Browse files
Files changed (1) hide show
  1. utils.py +2 -4
utils.py CHANGED
@@ -8,16 +8,14 @@ ENVIRONMENT = os.getenv("ENVIRONMENT")
8
 
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",
16
  }
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
  }
 
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
+ "Huggingface / Mistral 7B Instruct": "mistralai/Mistral-7B-Instruct-v0.3",
 
12
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
13
  "IBM Granite3.1 dense / Ollama local": "ollama",
14
  "Open AI / GPT-4o-mini": "openai",
15
  }
16
  else:
17
  AI_MODELS = {
18
+ "Huggingface / Mistral 7B Instruct": "mistralai/Mistral-7B-Instruct-v0.3",
 
19
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
20
  "Open AI / GPT-4o-mini": "openai",
21
  }