From bf0c2f2d3ad3f18c126005aed73a03781733c94d Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Mon, 5 Aug 2024 22:49:46 -0400 Subject: [PATCH] Update .buildkite/lm-eval-harness/test_lm_eval_correctness.py Co-authored-by: Cyrus Leung --- .buildkite/lm-eval-harness/test_lm_eval_correctness.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/.buildkite/lm-eval-harness/test_lm_eval_correctness.py b/.buildkite/lm-eval-harness/test_lm_eval_correctness.py index e1338a7443ca8..af3226f51f4ff 100644 --- a/.buildkite/lm-eval-harness/test_lm_eval_correctness.py +++ b/.buildkite/lm-eval-harness/test_lm_eval_correctness.py @@ -23,10 +23,7 @@ def launch_lm_eval(eval_config): - try: - trust_remote_code = eval_config['trust_remote_code'] - except Exception: - trust_remote_code = False + trust_remote_code = eval_config.get('trust_remote_code', False) model_args = f"pretrained={eval_config['model_name']}," \ f"tensor_parallel_size={TP_SIZE}," \