We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c8d53b3 commit 75972edCopy full SHA for 75972ed
haystack/components/readers/extractive.py
@@ -602,7 +602,8 @@ def run(
602
cur_input_ids = input_ids[start_index:end_index]
603
cur_attention_mask = attention_mask[start_index:end_index]
604
605
- output = self.model(input_ids=cur_input_ids, attention_mask=cur_attention_mask)
+ with torch.inference_mode():
606
+ output = self.model(input_ids=cur_input_ids, attention_mask=cur_attention_mask)
607
cur_start_logits = output.start_logits
608
cur_end_logits = output.end_logits
609
if num_batches != 1:
0 commit comments