datacipen commited on
Commit
a33086a
·
verified ·
1 Parent(s): e395fff

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -33,7 +33,8 @@ async def LLMistral():
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.2"
 
37
  #repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
38
  #repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"
39
  llm = HuggingFaceEndpoint(
 
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.2"
37
+ repo_id = "google/gemma-2-2b-it"
38
  #repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
39
  #repo_id = "allenai/OLMoE-1B-7B-0924-Instruct"
40
  llm = HuggingFaceEndpoint(