From 6eee83f5e382b7e9c607346916341a18a81b1d14 Mon Sep 17 00:00:00 2001 From: KevKibe Date: Wed, 11 Sep 2024 19:20:33 +0300 Subject: [PATCH] fix(eval): add trust_remote_code=True parameter to load_dataset implementation --- training/run_eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/training/run_eval.py b/training/run_eval.py index a92716f..e89da78 100644 --- a/training/run_eval.py +++ b/training/run_eval.py @@ -519,6 +519,7 @@ def main(): cache_dir=data_args.dataset_cache_dir, streaming=data_args.streaming, num_proc=data_args.preprocessing_num_workers, + trust_remote_code=True ) if data_args.only_short_form: