fixed typo. (#37057)

Signed-off-by: zhanluxianshen <zhanluxianshen@163.com>
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
This commit is contained in:
湛露先生
2025-03-29 01:12:14 +08:00
committed by GitHub
parent 2bea6bf24e
commit 257bc670fb
2 changed files with 2 additions and 2 deletions

View File

@@ -1102,7 +1102,7 @@ class SequenceBiasLogitsProcessor(LogitsProcessor):
self._convert_list_arguments_into_dict() self._convert_list_arguments_into_dict()
# Bias variables that will be populated on the first call (for retrocompatibility purposes, the vocabulary size # Bias variables that will be populated on the first call (for retrocompatibility purposes, the vocabulary size
# is infered in the first usage, which inhibits initializing here) # is inferred in the first usage, which inhibits initializing here)
self.length_1_bias = None self.length_1_bias = None
self.prepared_bias_variables = False self.prepared_bias_variables = False

View File

@@ -483,7 +483,7 @@ class TFGenerationMixin:
) -> tf.Tensor: ) -> tf.Tensor:
""" """
Computes the transition scores of sequences given the generation scores (and beam indices, if beam search was Computes the transition scores of sequences given the generation scores (and beam indices, if beam search was
used). This is a convenient method to quicky obtain the scores of the selected tokens at generation time. used). This is a convenient method to quickly obtain the scores of the selected tokens at generation time.
Parameters: Parameters:
sequences (`tf.Tensor`): sequences (`tf.Tensor`):