We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 81ca43a commit 74b7262Copy full SHA for 74b7262
examples/programmatic/post_process_topk.py
@@ -1,8 +1,6 @@
1
-from transformer_lens import hook_points
2
-from lm_saes import post_process_topk_to_jumprelu_runner, LanguageModelSAERunnerConfig, SAEConfig
3
import os
4
-import torch
5
-import jsonlines
+from lm_saes import post_process_topk_to_jumprelu_runner, LanguageModelSAERunnerConfig, SAEConfig
+
6
7
layer = 15
8
@@ -34,7 +32,7 @@
34
32
dict(
35
33
**sae_config,
36
model_name=model_name,
37
- model_from_pretrained_path=model_from_pretrained_path,
+ # model_from_pretrained_path=model_from_pretrained_path,
38
# d_model=4096,
39
dataset_path="<local_dataset_path>",
40
is_dataset_tokenized=False,
0 commit comments