Mixtral-PDF-Chat

Sleeping

Waflon commited on Jan 16

Commit

6b53f9c

•

1 Parent(s): 8c29218

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,8 +46,8 @@ def get_vectorstore(text_chunks : list) -> FAISS:
 def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
     # llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
     llm = HuggingFaceHub(
-        #repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
-        repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
         model_kwargs={"temperature": 0.5, "max_length": 1048},
     )

 def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
     # llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
     llm = HuggingFaceHub(
+        repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        #repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
         model_kwargs={"temperature": 0.5, "max_length": 1048},
     )