Skip to content

GitLab

Unverified Commit 2b8599b2 authored by Zhen Wang's avatar Zhen Wang Committed by GitHub
Browse files

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

parent f52746d0
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/transformers/pipelines/question_answering.py
View file @ 2b8599b2
...@@ -182,6 +182,8 @@ class QuestionAnsweringPipeline(ChunkPipeline): ...@@ -182,6 +182,8 @@ class QuestionAnsweringPipeline(ChunkPipeline):
preprocess_params["doc_stride"] = doc_stride preprocess_params["doc_stride"] = doc_stride
if max_question_len is not None: if max_question_len is not None:
preprocess_params["max_question_len"] = max_question_len 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 = {} postprocess_params = {}
if topk is not None and top_k is None: if topk is not None and top_k is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment