Carlosito16 commited on
Commit
8530356
1 Parent(s): d9c25e6

Update pages/2_model.py

Browse files
Files changed (1) hide show
  1. pages/2_model.py +1 -1
pages/2_model.py CHANGED
@@ -64,7 +64,7 @@ if load_model_button:
64
  st.session_state['temperature'] = temperature
65
  st.session_state['repetition_penalty'] = repetition_penalty
66
 
67
- # st.session_state['model'] = load_llm_model(max_length, temperature, repetition_penalty)
68
 
69
 
70
  st.write("⚠️ Please expect to wait **1 - 2 minutes ** for the application to download the 3-billion-parameter LLM")
 
64
  st.session_state['temperature'] = temperature
65
  st.session_state['repetition_penalty'] = repetition_penalty
66
 
67
+ st.session_state['model'] = load_llm_model(max_length, temperature, repetition_penalty)
68
 
69
 
70
  st.write("⚠️ Please expect to wait **1 - 2 minutes ** for the application to download the 3-billion-parameter LLM")