From 59bbdb81bf3fa6d05abff3470624a08063b7ea60 Mon Sep 17 00:00:00 2001
From: James Chua <chuajamessh@gmail.com>
Date: Sat, 6 May 2023 15:41:04 +0800
Subject: [PATCH] Revert "print when caching"

This reverts commit 363d8067ad1398fb834e617272f1c83807dc0b12.
---
 elk/extraction/extraction.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/elk/extraction/extraction.py b/elk/extraction/extraction.py
index 59646e79..53d02ee6 100644
--- a/elk/extraction/extraction.py
+++ b/elk/extraction/extraction.py
@@ -307,7 +307,6 @@ def extract_hiddens(
                 if is_enc_dec:
                     inputs["labels"] = answer
                 if cached_question_kv is not None:
-                    print("Using cached question key-values")
                     # If we cached the question, all we need to pass is the answer
                     inputs["past_key_values"] = cached_question_kv
                     inputs["input_ids"] = answer