From 14a416c1a03623ab3cf2ea2649abe50258d9eb18 Mon Sep 17 00:00:00 2001 From: LuyaoZhuang Date: Mon, 27 Oct 2025 01:19:38 -0400 Subject: Update --- run.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'run.py') diff --git a/run.py b/run.py index bed9f1a..b1f21f2 100644 --- a/run.py +++ b/run.py @@ -9,8 +9,7 @@ import warnings from src.evaluate import Evaluator from src.utils import LLM_Model from src.utils import setup_logging -os.environ["CUDA_VISIBLE_DEVICES"] = "5" -os.environ["TOKENIZERS_PARALLELISM"] = "false" + warnings.filterwarnings('ignore') def parse_arguments(): @@ -23,7 +22,7 @@ def parse_arguments(): return parser.parse_args() -def load_dataset(dataset_name,tokenizer): +def load_dataset(dataset_name): questions_path = f"dataset/{dataset_name}/questions.json" with open(questions_path, "r", encoding="utf-8") as f: questions = json.load(f) -- cgit v1.2.3