Generate: corrected exponential_decay_length_penalty type hint (huggi…

…ngface#19376)
ajsanjoaquin · Oct 12, 2022 · 20375d6 · 20375d6
1 parent 34b799b
commit 20375d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/generation_utils.py b/src/transformers/generation_utils.py
@@ -953,7 +953,7 @@ def generate(
         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,