Fix a bug that ignores max_seq_len in preprocess (#15238)

huggingface · Feb 14, 2022 · 2b8599b · 2b8599b
1 parent f52746d
commit 2b8599b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/pipelines/question_answering.py b/src/transformers/pipelines/question_answering.py
@@ -182,6 +182,8 @@ def _sanitize_parameters(
             preprocess_params["doc_stride"] = doc_stride
         if max_question_len is not None:
             preprocess_params["max_question_len"] = max_question_len
+        if max_seq_len is not None:
+            preprocess_params["max_seq_len"] = max_seq_len
 
         postprocess_params = {}
         if topk is not None and top_k is None: