Skip to content

GitLab

Unverified Commit 1ecedf1d authored by Austin's avatar Austin Committed by GitHub
Browse files

Update question_answering.py (#32208)

parent f53a5dec
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/pipelines/question_answering.py
View file @ 1ecedf1d
...@@ -118,7 +118,7 @@ def select_starts_ends( ...@@ -118,7 +118,7 @@ def select_starts_ends(
max_answer_len (`int`): Maximum size of the answer to extract from the model's output. max_answer_len (`int`): Maximum size of the answer to extract from the model's output.
""" """
# Ensure padded tokens & question tokens cannot belong to the set of candidate answers. # Ensure padded tokens & question tokens cannot belong to the set of candidate answers.
undesired_tokens = np.abs(p_mask.numpy() - 1) undesired_tokens = np.abs(np.array(p_mask) - 1)
if attention_mask is not None: if attention_mask is not None:
undesired_tokens = undesired_tokens & attention_mask undesired_tokens = undesired_tokens & attention_mask
......
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