ali121300 commited on
Commit
75201e2
1 Parent(s): ca48ed5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -52,10 +52,10 @@ def get_vectorstore(text_chunks : list) -> FAISS:
52
  def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
53
  # llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
54
  llm = HuggingFaceHub(
55
- repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF",
56
- #repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
57
  #repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
58
- model_kwargs={"temperature": 0.5, "max_length": 4096},
59
  )
60
 
61
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
52
  def get_conversation_chain(vectorstore:FAISS) -> ConversationalRetrievalChain:
53
  # llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613")
54
  llm = HuggingFaceHub(
55
+ repo_id="Tmistralai/Mistral-7B-Instruct-v0.2",
56
+ #repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1"
57
  #repo_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF"
58
+ model_kwargs={"temperature": 0.5, "max_length": 2048},
59
  )
60
 
61
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)