Fecalisboa commited on
Commit
503065c
·
verified ·
1 Parent(s): 716a46f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ import re
25
 
26
 
27
  # default_persist_directory = './chroma_HF/'
28
- list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
29
  "google/gemma-7b-it","google/gemma-2b-it", \
30
  "HuggingFaceH4/zephyr-7b-beta", "HuggingFaceH4/zephyr-7b-gemma-v0.1", \
31
  "meta-llama/Llama-2-7b-chat-hf", "microsoft/phi-2", \
@@ -102,7 +102,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
102
  # Use of trust_remote_code as model_kwargs
103
  # Warning: langchain issue
104
  # URL: https://github.com/langchain-ai/langchain/issues/6080
105
- if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.1":
106
  llm = HuggingFaceEndpoint(
107
  repo_id=llm_model,
108
  # model_kwargs={"temperature": temperature, "max_new_tokens": max_tokens, "top_k": top_k, "load_in_8bit": True}
 
25
 
26
 
27
  # default_persist_directory = './chroma_HF/'
28
+ list_llm = ["mistralai/Mistral-7B-Instruct-v0.3", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
29
  "google/gemma-7b-it","google/gemma-2b-it", \
30
  "HuggingFaceH4/zephyr-7b-beta", "HuggingFaceH4/zephyr-7b-gemma-v0.1", \
31
  "meta-llama/Llama-2-7b-chat-hf", "microsoft/phi-2", \
 
102
  # Use of trust_remote_code as model_kwargs
103
  # Warning: langchain issue
104
  # URL: https://github.com/langchain-ai/langchain/issues/6080
105
+ if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.3":
106
  llm = HuggingFaceEndpoint(
107
  repo_id=llm_model,
108
  # model_kwargs={"temperature": temperature, "max_new_tokens": max_tokens, "top_k": top_k, "load_in_8bit": True}