KvrParaskevi commited on
Commit
3cd7faf
1 Parent(s): 4b39069

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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, 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
 
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