Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit e4fe1ca

Browse files
committed
Ok can't remove import lm_eval. Move it inside eval condition
Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent 4778332 commit e4fe1ca

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

torchchat.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@
8282

8383
generate_main(args)
8484
elif args.command == "eval":
85+
import lm_eval # noqa
8586
from torchchat.usages.eval import main as eval_main
8687

8788
eval_main(args)

torchchat/cli/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
import torch
1515

1616
from torchchat.cli.download import download_and_convert, is_model_downloaded
17-
17+
import lm_eval # noqa
1818
from torchchat.utils.build_utils import (
1919
allowable_dtype_names,
2020
allowable_params_table,

0 commit comments

Comments
 (0)