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