Stefan Heimersheim commited on
Commit
bf227fc
1 Parent(s): dae8352

add all models and a warning

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,7 +13,7 @@ import circuitsvis as cv
13
  # Little bit of front end for model selector
14
 
15
  # Radio buttons
16
- model_name = st.sidebar.radio("Model", [
17
  "gelu-1l",
18
  "gelu-2l",
19
  "gelu-3l",
@@ -31,9 +31,9 @@ model_name = st.sidebar.radio("Model", [
31
  "solu-10l",
32
  "solu-12l",
33
  "gpt2-small",
34
- #"gpt2-medium",
35
- #"gpt2-large",
36
- #"gpt2-xl",
37
  ], index=1)
38
 
39
 
13
  # Little bit of front end for model selector
14
 
15
  # Radio buttons
16
+ model_name = st.sidebar.radio("Model (only use patching for\nsmall models due to memory limits)", [
17
  "gelu-1l",
18
  "gelu-2l",
19
  "gelu-3l",
31
  "solu-10l",
32
  "solu-12l",
33
  "gpt2-small",
34
+ "gpt2-medium",
35
+ "gpt2-large",
36
+ "gpt2-xl",
37
  ], index=1)
38
 
39