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="
|
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 |
)
|