Skip to content

GitLab

Unverified Commit c37dcff7 authored by Michael Glass's avatar Michael Glass Committed by GitHub
Browse files

Fixed parameter name for logits_processor (#9790)

parent 0d0efd3a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/rag/modeling_rag.py
View file @ c37dcff7
...@@ -1486,7 +1486,7 @@ class RagTokenForGeneration(RagPreTrainedModel): ...@@ -1486,7 +1486,7 @@ class RagTokenForGeneration(RagPreTrainedModel):
) )
return self.greedy_search( return self.greedy_search(
input_ids, input_ids,
pre_processor=pre_processor, logits_processor=pre_processor,
max_length=max_length, max_length=max_length,
pad_token_id=pad_token_id, pad_token_id=pad_token_id,
eos_token_id=eos_token_id, eos_token_id=eos_token_id,
...@@ -1509,7 +1509,7 @@ class RagTokenForGeneration(RagPreTrainedModel): ...@@ -1509,7 +1509,7 @@ class RagTokenForGeneration(RagPreTrainedModel):
return self.beam_search( return self.beam_search(
input_ids, input_ids,
beam_scorer, beam_scorer,
pre_processor=pre_processor, logits_processor=pre_processor,
max_length=max_length, max_length=max_length,
pad_token_id=pad_token_id, pad_token_id=pad_token_id,
eos_token_id=eos_token_id, eos_token_id=eos_token_id,
......
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