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

Commit cc37640

Browse files
committed
Bump lm_eval import before torchtune
1 parent a0a92ba commit cc37640

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

torchchat/usages/eval.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,10 @@
33

44
# This source code is licensed under the license found in the
55
# LICENSE file in the root directory of this source tree.
6-
import argparse
76
from typing import Callable, Optional
87

8+
import lm_eval
9+
910
import torch
1011
import torch._dynamo.config
1112
import torch._inductor.config
@@ -16,7 +17,6 @@
1617
BuilderArgs,
1718
TokenizerArgs,
1819
)
19-
from torchchat.cli.cli import add_arguments_for_verb, arg_init
2020

2121
from torchchat.model import Model
2222
from torchchat.utils.build_utils import set_precision
@@ -28,8 +28,6 @@
2828
torch._inductor.config.triton.cudagraphs = True
2929
torch._dynamo.config.cache_size_limit = 100000
3030

31-
import lm_eval
32-
3331
from lm_eval.evaluator import evaluate
3432
from lm_eval.models.huggingface import HFLM as eval_wrapper
3533
from lm_eval.tasks import get_task_dict

0 commit comments

Comments
 (0)