Chris4K commited on
Commit
3731d17
·
1 Parent(s): 273c375

chat to msg

Browse files
model/conversation_chain_singleton.py CHANGED
@@ -56,7 +56,7 @@ class ConversationChainSingleton:
56
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
57
  model_kwargs={"max_length": 1048, "temperature": 0.2, "max_new_tokens": 256, "top_p": 0.95, "repetition_penalty": 1.0},
58
  )
59
- memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
60
  conversation_chain = ConversationChain(
61
  llm=llm, verbose=True, memory=memory
62
  )
 
56
  repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
57
  model_kwargs={"max_length": 1048, "temperature": 0.2, "max_new_tokens": 256, "top_p": 0.95, "repetition_penalty": 1.0},
58
  )
59
+ memory = ConversationBufferMemory(memory_key="history", return_messages=True)
60
  conversation_chain = ConversationChain(
61
  llm=llm, verbose=True, memory=memory
62
  )