Update model.py
Browse files
model.py
CHANGED
|
@@ -53,7 +53,7 @@ def chat_template_prompt():
|
|
| 53 |
def chain():
|
| 54 |
#memory = ConversationBufferMemory(memory_key="history")
|
| 55 |
chat_prompt = chat_template_prompt()
|
| 56 |
-
memory = ConversationBufferWindowMemory(k=3
|
| 57 |
llm_chain = LLMChain(llm=llm, memory = memory, prompt = chat_prompt)
|
| 58 |
memory.load_memory_variables({}) #Initialize memory
|
| 59 |
return llm_chain
|
|
|
|
| 53 |
def chain():
|
| 54 |
#memory = ConversationBufferMemory(memory_key="history")
|
| 55 |
chat_prompt = chat_template_prompt()
|
| 56 |
+
memory = ConversationBufferWindowMemory(k=3) #memory_key="history"
|
| 57 |
llm_chain = LLMChain(llm=llm, memory = memory, prompt = chat_prompt)
|
| 58 |
memory.load_memory_variables({}) #Initialize memory
|
| 59 |
return llm_chain
|