From c35fd34ca6b5bcd2ae068dc94533e561e399dbc9 Mon Sep 17 00:00:00 2001 From: Langfun Authors Date: Mon, 10 Nov 2025 16:21:22 -0800 Subject: [PATCH] Internal PiperOrigin-RevId: 830641606 --- langfun/core/eval/v2/checkpointing.py | 2 +- langfun/core/eval/v2/runners.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/langfun/core/eval/v2/checkpointing.py b/langfun/core/eval/v2/checkpointing.py index 600ddc93..5cac86ef 100644 --- a/langfun/core/eval/v2/checkpointing.py +++ b/langfun/core/eval/v2/checkpointing.py @@ -159,7 +159,7 @@ def _load_state(ckpt_file): _ = list( lf.concurrent_map( - _load_state, ckpt_files, max_workers=16, silence_on_errors=None + _load_state, ckpt_files, max_workers=128, silence_on_errors=None ) ) diff --git a/langfun/core/eval/v2/runners.py b/langfun/core/eval/v2/runners.py index 78194e28..dac41bef 100644 --- a/langfun/core/eval/v2/runners.py +++ b/langfun/core/eval/v2/runners.py @@ -73,7 +73,7 @@ def _on_bound(self): self.plugins.append(progress_tracking.progress_tracker(self.tqdm)) self._io_pool_lock = threading.Lock() - self._io_pool = concurrent.futures.ThreadPoolExecutor(max_workers=16) + self._io_pool = concurrent.futures.ThreadPoolExecutor(max_workers=128) # TODO(daiyip): render background errors. self._background_last_error = None