Skip to content

Commit 2a1409d

Browse files
authored
move environment variable setting into eval function (#829)
1 parent c898d13 commit 2a1409d

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

auto_round/script/llm.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,6 @@
4242
set_cuda_visible_devices,
4343
)
4444

45-
os.environ["TOKENIZERS_PARALLELISM"] = "false"
46-
4745

4846
class BasicArgumentParser(argparse.ArgumentParser):
4947

@@ -978,12 +976,12 @@ def eval_task_by_task(
978976
def eval_with_vllm(args):
979977
import time
980978

981-
st = time.time()
982-
983979
from lm_eval import evaluator # pylint: disable=E0401
984980
from lm_eval.models.vllm_causallms import VLLM # pylint: disable=E0401
985981
from lm_eval.utils import make_table # pylint: disable=E0401
986982

983+
st = time.time()
984+
os.environ["TOKENIZERS_PARALLELISM"] = "false"
987985
device_str, _ = get_device_and_parallelism(args.device)
988986
eval_model_dtype = get_model_dtype(args.eval_model_dtype, "auto")
989987
if (batch_size := args.eval_bs) is None:

0 commit comments

Comments
 (0)