Add layer sweep models

etowahadams · Jan 17, 2025 · 663361b · 663361b
1 parent 0c02561
commit 663361b
Showing 1 changed file with 46 additions and 1 deletion.
diff --git a/viz/src/SAEConfigs.ts b/viz/src/SAEConfigs.ts
@@ -414,8 +414,53 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
     defaultDim: 4000,
     supportsCustomSequence: true,
   },
+  "SAE4096-L8": {
+    storagePath: "4096_layer_sweep/esm2_plm1280_l8_sae4096_aux640",
+    description: "",
+    numHiddenDims: 4096,
+    plmLayer: 8,
+    defaultDim: 0,
+    supportsCustomSequence: false,
+    curated: [],
+  },
+  "SAE4096-L12": {
+    storagePath: "4096_layer_sweep/esm2_plm1280_l12_sae4096_aux640",
+    description: "",
+    numHiddenDims: 4096,
+    plmLayer: 12,
+    defaultDim: 0,
+    supportsCustomSequence: false,
+    curated: [],
+  },
+  "SAE4096-L16": {
+    storagePath: "4096_layer_sweep/esm2_plm1280_l16_sae4096_aux640",
+    description: "",
+    numHiddenDims: 4096,
+    plmLayer: 16,
+    defaultDim: 0,
+    supportsCustomSequence: false,
+    curated: [],
+  },
+  "SAE4096-L28": {
+    storagePath: "4096_layer_sweep/esm2_plm1280_l28_sae4096_aux640",
+    description: "",
+    numHiddenDims: 4096,
+    plmLayer: 28,
+    defaultDim: 0,
+    supportsCustomSequence: false,
+    curated: [],
+  },
+  "SAE4096-L32": {
+    storagePath: "4096_layer_sweep/esm2_plm1280_l32_sae4096_aux640",
+    description: "",
+    numHiddenDims: 4096,
+    plmLayer: 32,
+    defaultDim: 0,
+    supportsCustomSequence: false,
+    curated: [],
+  },
   "SAE4096-L33": {
-    storagePath: "esm2_plm1280_l33_sae4096_aux640",
+    storagePath: "4096_layer_sweep/esm2_plm1280_l33_sae4096_aux640",
     description: "",
     numHiddenDims: 4096,
     plmLayer: 33,