From 1a85299a5e4e9df66daa6e28441023b40c46ed8d Mon Sep 17 00:00:00 2001 From: Siddharth Jain <35868478+SidJain1412@users.noreply.github.com> Date: Tue, 15 Sep 2020 17:48:42 +0530 Subject: [PATCH] Tiny typo fix (#7143) --- src/transformers/generation_tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/generation_tf_utils.py b/src/transformers/generation_tf_utils.py index b6c5a596ab..e781cc1ae2 100644 --- a/src/transformers/generation_tf_utils.py +++ b/src/transformers/generation_tf_utils.py @@ -96,7 +96,7 @@ class TFGenerationMixin: Whether to stop the beam search when at least ``num_beams`` sentences are finished per batch or not. num_beams (:obj:`int`, `optional`, defaults to 1): Number of beams for beam search. 1 means no beam search. - temperature (:obj:`float`, `optional`, defaults tp 1.0): + temperature (:obj:`float`, `optional`, defaults to 1.0): The value used to module the next token probabilities. top_k (:obj:`int`, `optional`, defaults to 50): The number of highest probability vocabulary tokens to keep for top-k-filtering.