Spaces:
Running
Running
Commit
·
f318e1a
1
Parent(s):
50c004c
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ if 'memory' not in st.session_state:
|
|
29 |
|
30 |
if 'chain' not in st.session_state:
|
31 |
st.session_state['chain'] = custom_chain_with_history(
|
32 |
-
llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"]),
|
33 |
# memory=st.session_state.memory.chat_memory,
|
34 |
memory=st.session_state.memory
|
35 |
)
|
|
|
29 |
|
30 |
if 'chain' not in st.session_state:
|
31 |
st.session_state['chain'] = custom_chain_with_history(
|
32 |
+
llm=CustomLLM(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", model_type='text-generation', api_token=API_TOKEN, stop=["\n<|","<|"], temperature=0.001),
|
33 |
# memory=st.session_state.memory.chat_memory,
|
34 |
memory=st.session_state.memory
|
35 |
)
|