TomData commited on
Commit
51e3323
·
verified ·
1 Parent(s): 76a59b7

Update src/chatbot.py

Browse files
Files changed (1) hide show
  1. src/chatbot.py +1 -1
src/chatbot.py CHANGED
@@ -21,7 +21,7 @@ load_dotenv(find_dotenv())
21
  embeddings = HuggingFaceEmbeddings(model_name="paraphrase-multilingual-MiniLM-L12-v2") # Remove embedding input parameter from functions?
22
  llm = HuggingFaceHub(
23
  # ToDo: Try different models here
24
- repo_id="mistralai/Mistral-7B-Instruct-v0.3"
25
  #repo_id = "mistralai/Ministral-8B-Instruct-2410"
26
  #repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
27
  # repo_id="CohereForAI/c4ai-command-r-v01", # too large 69gb
 
21
  embeddings = HuggingFaceEmbeddings(model_name="paraphrase-multilingual-MiniLM-L12-v2") # Remove embedding input parameter from functions?
22
  llm = HuggingFaceHub(
23
  # ToDo: Try different models here
24
+ repo_id="mistralai/Mistral-7B-Instruct-v0.3",
25
  #repo_id = "mistralai/Ministral-8B-Instruct-2410"
26
  #repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
27
  # repo_id="CohereForAI/c4ai-command-r-v01", # too large 69gb