Spaces:

StefanHex
/

simple-trafo-mech-int

Runtime error

Stefan Heimersheim commited on Jan 23, 2023

Commit

dae8352

1 Parent(s): e4ba3cc

Try more models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,21 +16,21 @@ import circuitsvis as cv
 model_name = st.sidebar.radio("Model", [
     "gelu-1l",
     "gelu-2l",
-    #"gelu-3l",
-    #"gelu-4l",
-    #"attn-only-1l",
-    #"attn-only-2l",
-    #"attn-only-3l",
-    #"attn-only-4l",
-    #"solu-1l",
-    #"solu-2l",
-    #"solu-3l",
-    #"solu-4l",
-    #"solu-6l",
-    #"solu-8l",
-    #"solu-10l",
-    #"solu-12l",
-    #"gpt2-small",
     #"gpt2-medium",
     #"gpt2-large",
     #"gpt2-xl",

 model_name = st.sidebar.radio("Model", [
     "gelu-1l",
     "gelu-2l",
+    "gelu-3l",
+    "gelu-4l",
+    "attn-only-1l",
+    "attn-only-2l",
+    "attn-only-3l",
+    "attn-only-4l",
+    "solu-1l",
+    "solu-2l",
+    "solu-3l",
+    "solu-4l",
+    "solu-6l",
+    "solu-8l",
+    "solu-10l",
+    "solu-12l",
+    "gpt2-small",
     #"gpt2-medium",
     #"gpt2-large",
     #"gpt2-xl",