File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 4242 set_cuda_visible_devices ,
4343)
4444
45- os .environ ["TOKENIZERS_PARALLELISM" ] = "false"
46-
4745
4846class BasicArgumentParser (argparse .ArgumentParser ):
4947
@@ -978,12 +976,12 @@ def eval_task_by_task(
978976def 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 :
You can’t perform that action at this time.
0 commit comments