add logs

summarize.py

@@ -1,3 +1,5 @@
+import logging
+
 import torch
 from tqdm.auto import tqdm
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
@@ -88,7 +90,8 @@ def summarize_via_tokenbatches(
     Returns:
         str: the summary
     """
-
+    # log all input parameters
+    logging.info(f"input parameters: {kwargs}, batch_length={batch_length}, batch_stride={batch_stride}")
     encoded_input = tokenizer(
         input_text,
         padding="max_length",