SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fixed parameter name for logits_processor (#9790)

Browse Source
This commit is contained in:
Michael Glass
2021-01-26 10:44:02 -05:00
committed by GitHub
parent 0d0efd3a0e
commit c37dcff764
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/rag/modeling_rag.py
View File

@@ -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):
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,