Revert "print when caching"

This reverts commit 363d806.
EleutherAI · May 6, 2023 · 59bbdb8 · 59bbdb8
1 parent 363d806
commit 59bbdb8
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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