Changed info to debug logging for cache.

huggingface · Dec 30, 2024 · cb16cc5 · cb16cc5
1 parent 6a9bb65
commit cb16cc5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lighteval/models/transformers/base_model.py b/src/lighteval/models/transformers/base_model.py
@@ -872,7 +872,7 @@ def _generate_cached(
         # Try to get from cache first
         cached_responses = self._generation_cache.get(cache_key)
         if cached_responses is not None:
-            logger.info(f"Cache hit for batch {cache_key}")
+            logger.debug(f"Cache hit for batch {cache_key}")
             return cached_responses
 
         # Generate if not in cache
@@ -886,7 +886,7 @@ def _generate_cached(
         )
 
         # Cache the results
-        logger.info(f"Caching results for batch {cache_key}")
+        logger.debug(f"Caching results for batch {cache_key}")
         self._generation_cache.set(cache_key, cur_reponses)
 
         return cur_reponses