datacipen commited on
Commit
64e3aee
1 Parent(s): ed13a2f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -30,8 +30,8 @@ from datetime import timedelta
30
  @cl.step(type="tool")
31
  async def LLMistral():
32
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
33
- repo_id = "mistralai/Ministral-8B-Instruct-2410"
34
- #repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
35
  #repo_id = "nvidia/Mistral-NeMo-Minitron-8B-Base"
36
  #repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
37
  #repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"
 
30
  @cl.step(type="tool")
31
  async def LLMistral():
32
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
33
+ #repo_id = "mistralai/Ministral-8B-Instruct-2410"
34
+ repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
35
  #repo_id = "nvidia/Mistral-NeMo-Minitron-8B-Base"
36
  #repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
37
  #repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"