jonathanjordan21 commited on
Commit
f318e1a
1 Parent(s): 50c004c

Update app.py

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