Hellisotherpeople commited on
Commit
71d0e66
β€’
1 Parent(s): 90ed380

Update Text-Generation.py

Browse files
Files changed (1) hide show
  1. Text-Generation.py +1 -1
Text-Generation.py CHANGED
@@ -49,7 +49,7 @@ st.caption("The inspiration for this space: https://en.wikipedia.org/wiki/Gadsby
49
  form = st.sidebar.form("choose_settings")
50
  form.header("Model Settings")
51
 
52
- model_name = form.text_area("Enter the name of the pre-trained model from transformers that we are using for Text Generation", value = "facebook/opt-1.3b")
53
  form.caption("This will download a new model, so it may take awhile or even break if the model is too large")
54
  percision = form.selectbox("What percision are we loading the model with?", ["8bit", "16bit", "32bit"], )
55
  form.caption("The lower the percision, the less ram the model takes and the faster it runs, but the quality is reduced")
49
  form = st.sidebar.form("choose_settings")
50
  form.header("Model Settings")
51
 
52
+ model_name = form.text_area("Enter the name of the pre-trained model from transformers that we are using for Text Generation", value = "TheBloke/vicuna-7B-1.1-HF")
53
  form.caption("This will download a new model, so it may take awhile or even break if the model is too large")
54
  percision = form.selectbox("What percision are we loading the model with?", ["8bit", "16bit", "32bit"], )
55
  form.caption("The lower the percision, the less ram the model takes and the faster it runs, but the quality is reduced")