Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
| 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.
|
| 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}
|