Generate: corrected exponential_decay_length_penalty type hint (#19376)

d866b485 · Shivang Mishra · GitHub · 4dd784c3 · d866b485
Unverified Commit d866b485 authored Oct 10, 2022 by Shivang Mishra Committed by GitHub Oct 10, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/generation_utils.py src/transformers/generation_utils.py +1 -1

No files found.
--- a/src/transformers/generation_utils.py
+++ b/src/transformers/generation_utils.py
@@ -953,7 +953,7 @@ class GenerationMixin:
        forced_eos_token_id: Optional[int] = None,
        remove_invalid_values: Optional[bool] = None,
        synced_gpus: Optional[bool] = False,
-        exponential_decay_length_penalty: Optional[Tuple[Union[int, float]]] = None,
+        exponential_decay_length_penalty: Optional[Tuple[int, float]] = None,
        suppress_tokens: Optional[List[int]] = None,
        begin_suppress_tokens: Optional[List[int]] = None,
        forced_decoder_ids: Optional[List[int]] = None,