riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Small fix to ExponentialDecayLengthPenalty docstring (#21308)

Browse Source 
Currently, it incorrectly states that the exponential_decay_length_penalty tuple parameter is optional.

Also changed the corresponding type hint to be more specific.
This commit is contained in:
Nick Hill 2023-01-25 11:46:08 -08:00 committed by GitHub
parent 3a6e4a221c
commit 140c6edeb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/generation/logits_process.py
View File

@ -825,7 +825,7 @@ class ExponentialDecayLengthPenalty(LogitsProcessor):
reached.
Args:
exponential_decay_length_penalty (`tuple(int, float)`, *optional*):
exponential_decay_length_penalty (`tuple(int, float)`):
This tuple shall consist of: `(start_index, decay_factor)` where `start_index` indicates where penalty
starts and `decay_factor` represents the factor of exponential decay
eos_token_id (`Union[int, List[int]]`):
@ -835,7 +835,10 @@ class ExponentialDecayLengthPenalty(LogitsProcessor):
"""
def __init__(
self, exponential_decay_length_penalty: Tuple, eos_token_id: Union[int, List[int]], input_ids_seq_length: int
self,
exponential_decay_length_penalty: Tuple[int, float],
eos_token_id: Union[int, List[int]],
input_ids_seq_length: int,
):
self.regulation_start = exponential_decay_length_penalty[0] + input_ids_seq_length
self.regulation_factor = exponential_decay_length_penalty[1]