Skip to content

Commit 82056f8

Browse files
author
Junxuan Wang
committed
fix post_process_topk.py
1 parent 751c740 commit 82056f8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/programmatic/post_process_topk.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
1-
import os
21
import argparse
2+
import os
33

4-
from lm_saes import LanguageModelSAERunnerConfig, SAEConfig, post_process_topk_to_jumprelu_runner
4+
from lm_saes import (LanguageModelSAERunnerConfig, SAEConfig,
5+
post_process_topk_to_jumprelu_runner)
56

67
parser = argparse.ArgumentParser(description="Process hyparameters")
78
parser.add_argument("-l", "--layer", nargs="*", required=False, type=int, help="Layer number")

0 commit comments

Comments
 (0)