Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -137,12 +137,18 @@ def get_vectorstore(text_chunks):
|
|
137 |
|
138 |
def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
|
139 |
# llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
|
|
|
|
|
|
|
|
|
|
|
|
|
140 |
llm = HuggingFaceHub(
|
141 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
142 |
-
|
143 |
-
#repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
|
144 |
-
model_kwargs={"temperature": 0.5, "max_length": 2096},#1048
|
145 |
)
|
|
|
|
|
146 |
|
147 |
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
148 |
conversation_chain = ConversationalRetrievalChain.from_llm(
|
|
|
137 |
|
138 |
def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
|
139 |
# llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
|
140 |
+
#llm = HuggingFaceHub(
|
141 |
+
# repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
142 |
+
# #repo_id="clibrain/lince-mistral-7b-it-es",
|
143 |
+
# #repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
|
144 |
+
# model_kwargs={"temperature": 0.5, "max_length": 2096},#1048
|
145 |
+
#)
|
146 |
llm = HuggingFaceHub(
|
147 |
repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
|
148 |
+
model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "max_length": 1048, "top_k": 3, "trust_remote_code": True, "torch_dtype": "auto"},
|
|
|
|
|
149 |
)
|
150 |
+
# llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
|
151 |
+
|
152 |
|
153 |
memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
154 |
conversation_chain = ConversationalRetrievalChain.from_llm(
|