increase loader cache

Signed-off-by: dafnapension <[email protected]>
IBM · Jan 22, 2025 · 6080c1e · 6080c1e
1 parent a375de0
commit 6080c1e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/performance/bluebench_profiler.py b/performance/bluebench_profiler.py
@@ -62,7 +62,6 @@ def profiler_generate_benchmark_dataset(
         with settings.context(
             disable_hf_datasets_cache=False,
             allow_unverified_code=True,
-            mock_inference_mode=True,
         ):
             stream = benchmark_recipe()[split]
 

diff --git a/src/unitxt/settings_utils.py b/src/unitxt/settings_utils.py
@@ -150,7 +150,7 @@ def __getattr__(self, key):
     settings.data_classification_policy = None
     settings.mock_inference_mode = (bool, False)
     settings.disable_hf_datasets_cache = (bool, True)
-    settings.loader_cache_size = (int, 1)
+    settings.loader_cache_size = (int, 25)
     settings.task_data_as_text = (bool, True)
     settings.default_provider = "watsonx"
     settings.default_format = None