Attention vis works up to gpt2-medium

app.py

@@ -13,7 +13,7 @@ import circuitsvis as cv
 # Little bit of front end for model selector
 
 # Radio buttons
-model_name = st.sidebar.radio("Model (only use patching for\nsmall models due to memory limits)", [
+model_name = st.sidebar.radio("Model (only use patching for\nsmall (<4L) models due to memory limits)", [
     "gelu-1l",
     "gelu-2l",
     "gelu-3l",
@@ -32,8 +32,8 @@ model_name = st.sidebar.radio("Model (only use patching for\nsmall models due to
     "solu-12l",
     "gpt2-small",
     "gpt2-medium",
-    "gpt2-large",
-    "gpt2-xl",
+    #"gpt2-large",
+    #"gpt2-xl",
     ], index=1)