Spaces:
Running
Running
Commit
•
4ed8fad
1
Parent(s):
9cbe2ec
Update app.py
Browse files
app.py
CHANGED
@@ -26,8 +26,8 @@ POSTGRE_URL = os.environ['POSTGRE_URL']
|
|
26 |
def get_llm_chain():
|
27 |
return custom_chain_with_history(
|
28 |
llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
|
29 |
-
|
30 |
-
memory=st.session_state.memory
|
31 |
)
|
32 |
|
33 |
|
|
|
26 |
def get_llm_chain():
|
27 |
return custom_chain_with_history(
|
28 |
llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
|
29 |
+
memory=st.session_state.memory.chat_memory,
|
30 |
+
# memory=st.session_state.memory
|
31 |
)
|
32 |
|
33 |
|