Unverified Commit 1a85299a authored by Siddharth Jain's avatar Siddharth Jain Committed by GitHub
Browse files

Tiny typo fix (#7143)

parent e29c3f1b
...@@ -96,7 +96,7 @@ class TFGenerationMixin: ...@@ -96,7 +96,7 @@ class TFGenerationMixin:
Whether to stop the beam search when at least ``num_beams`` sentences are finished per batch or not. 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): num_beams (:obj:`int`, `optional`, defaults to 1):
Number of beams for beam search. 1 means no beam search. 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. The value used to module the next token probabilities.
top_k (:obj:`int`, `optional`, defaults to 50): top_k (:obj:`int`, `optional`, defaults to 50):
The number of highest probability vocabulary tokens to keep for top-k-filtering. The number of highest probability vocabulary tokens to keep for top-k-filtering.
......
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