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

update mistral model

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -8,14 +8,14 @@ ENVIRONMENT = os.getenv("ENVIRONMENT")
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
  }
 
8
 
9
  if ENVIRONMENT == "dev":
10
  AI_MODELS = {
11
+ "Huggingface / Mistral 8B Instruct": "mistralai/Ministral-8B-Instruct-2410",
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 8B Instruct": "mistralai/Ministral-8B-Instruct-2410",
19
  "Huggingface / Meta Llama 3.1 8B Instruct": "meta-llama/Llama-3.1-8B-Instruct",
20
  "Open AI / GPT-4o-mini": "openai",
21
  }