From d866b4858a4e7b51b21f9ff623d8312528e73c49 Mon Sep 17 00:00:00 2001
From: Shivang Mishra <35092323+ShivangMishra@users.noreply.github.com>
Date: Mon, 10 Oct 2022 22:02:03 +0530
Subject: [PATCH] Generate: corrected exponential_decay_length_penalty type
 hint (#19376)

---
 src/transformers/generation_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/transformers/generation_utils.py b/src/transformers/generation_utils.py
index 380eec07270..27b392a23d5 100644
--- 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,