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 106f11a commit f7a1f11Copy full SHA for f7a1f11
eval.py
@@ -20,7 +20,7 @@
20
21
from build.model import Transformer
22
from build.utils import set_precision
23
-from cli import add_arguments_for_eval, arg_init
+from cli import add_arguments_for_verb, arg_init
24
from generate import encode_tokens, model_forward
25
26
torch._dynamo.config.automatic_dynamic_shapes = True
@@ -278,7 +278,7 @@ def main(args) -> None:
278
279
if __name__ == "__main__":
280
parser = argparse.ArgumentParser(description="torchchat eval CLI")
281
- add_arguments_for_eval(parser)
+ add_arguments_for_verb(parser, "eval")
282
args = parser.parse_args()
283
args = arg_init(args)
284
main(args)
0 commit comments