KvrParaskevi
commited on
Commit
•
3cd7faf
1
Parent(s):
4b39069
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
|