Pippoz commited on
Commit
9e2e1de
·
1 Parent(s): 0fffa24

Adding 2.7 Billion parameter model

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -31,7 +31,7 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('OPT-125m', 'OPT-350m', 'OPT-1.3b'), index = 1)
35
 
36
  if select_model == 'OPT-1.3b':
37
  model = 'facebook/opt-1.3b'
@@ -39,6 +39,9 @@ with col2:
39
  model = 'facebook/opt-350m'
40
  elif select_model == 'OPT-125m':
41
  model = 'facebook/opt-125m'
 
 
 
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('OPT-125m', 'OPT-350m', 'OPT-1.3b', 'OPT-2.7b'), index = 1)
35
 
36
  if select_model == 'OPT-1.3b':
37
  model = 'facebook/opt-1.3b'
 
39
  model = 'facebook/opt-350m'
40
  elif select_model == 'OPT-125m':
41
  model = 'facebook/opt-125m'
42
+ elif select_model == 'OPT-2.7b':
43
+ model = 'facebook/opt-2.7b'
44
+
45
 
46
  with st.spinner('Loading Model... (This may take a while)'):
47
  generator = get_model()