Skip to content

Commit f7a1f11

Browse files
committed
Fix loose reference
1 parent 106f11a commit f7a1f11

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

eval.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
from build.model import Transformer
2222
from build.utils import set_precision
23-
from cli import add_arguments_for_eval, arg_init
23+
from cli import add_arguments_for_verb, arg_init
2424
from generate import encode_tokens, model_forward
2525

2626
torch._dynamo.config.automatic_dynamic_shapes = True
@@ -278,7 +278,7 @@ def main(args) -> None:
278278

279279
if __name__ == "__main__":
280280
parser = argparse.ArgumentParser(description="torchchat eval CLI")
281-
add_arguments_for_eval(parser)
281+
add_arguments_for_verb(parser, "eval")
282282
args = parser.parse_args()
283283
args = arg_init(args)
284284
main(args)

0 commit comments

Comments
 (0)