From 0f3ad1507ecb181019ea5c6dc5c7beb43231a202 Mon Sep 17 00:00:00 2001 From: PenutChen Date: Thu, 22 Apr 2021 22:10:16 +0800 Subject: [PATCH] Fix typo (#11369) --- 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 165fc4aa12..60f7e14045 100644 --- a/src/transformers/generation_utils.py +++ b/src/transformers/generation_utils.py @@ -700,7 +700,7 @@ class GenerationMixin: 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.