Skip to content

Commit 74b7262

Browse files
author
zhengfuhe
committed
add post_process example
1 parent 81ca43a commit 74b7262

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/programmatic/post_process_topk.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
1-
from transformer_lens import hook_points
2-
from lm_saes import post_process_topk_to_jumprelu_runner, LanguageModelSAERunnerConfig, SAEConfig
31
import os
4-
import torch
5-
import jsonlines
2+
from lm_saes import post_process_topk_to_jumprelu_runner, LanguageModelSAERunnerConfig, SAEConfig
3+
64

75
layer = 15
86

@@ -34,7 +32,7 @@
3432
dict(
3533
**sae_config,
3634
model_name=model_name,
37-
model_from_pretrained_path=model_from_pretrained_path,
35+
# model_from_pretrained_path=model_from_pretrained_path,
3836
# d_model=4096,
3937
dataset_path="<local_dataset_path>",
4038
is_dataset_tokenized=False,

0 commit comments

Comments
 (0)