Skip to content

Commit 9f34ac6

Browse files
author
Junxuan Wang
committed
fix post_process_topk.py
1 parent 82056f8 commit 9f34ac6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/programmatic/post_process_topk.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
import argparse
22
import os
33

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

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

0 commit comments

Comments
 (0)