Skip to content

GitLab

Unverified Commit d866b485 authored by Shivang Mishra's avatar Shivang Mishra Committed by GitHub
Browse files

Generate: corrected exponential_decay_length_penalty type hint (#19376)

parent 4dd784c3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/generation_utils.py
View file @ d866b485
...@@ -953,7 +953,7 @@ class GenerationMixin: ...@@ -953,7 +953,7 @@ class GenerationMixin:
forced_eos_token_id: Optional[int] = None, forced_eos_token_id: Optional[int] = None,
remove_invalid_values: Optional[bool] = None, remove_invalid_values: Optional[bool] = None,
synced_gpus: Optional[bool] = False, 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, suppress_tokens: Optional[List[int]] = None,
begin_suppress_tokens: Optional[List[int]] = None, begin_suppress_tokens: Optional[List[int]] = None,
forced_decoder_ids: Optional[List[int]] = None, forced_decoder_ids: Optional[List[int]] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment