diff --git a/viz/src/SAEConfigs.ts b/viz/src/SAEConfigs.ts index 14472cc..c320a26 100644 --- a/viz/src/SAEConfigs.ts +++ b/viz/src/SAEConfigs.ts @@ -414,8 +414,53 @@ export const SAE_CONFIGS: Record = { 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,