Spaces:

Carlosito16
/

extraGPT

Paused

Carlosito16 commited on Aug 28, 2023

Commit

8530356

•

1 Parent(s): d9c25e6

Update pages/2_model.py

Files changed (1) hide show

pages/2_model.py CHANGED Viewed

@@ -64,7 +64,7 @@ if load_model_button:
     st.session_state['temperature'] = temperature
     st.session_state['repetition_penalty'] = repetition_penalty
-    # st.session_state['model'] = load_llm_model(max_length, temperature, repetition_penalty)
     st.write("⚠️ Please expect to wait **1 - 2 minutes **  for the application to download the 3-billion-parameter LLM")

     st.session_state['temperature'] = temperature
     st.session_state['repetition_penalty'] = repetition_penalty
+    st.session_state['model'] = load_llm_model(max_length, temperature, repetition_penalty)
     st.write("⚠️ Please expect to wait **1 - 2 minutes **  for the application to download the 3-billion-parameter LLM")