fixed typo. (#37057)
Signed-off-by: zhanluxianshen <zhanluxianshen@163.com> Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
@@ -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`):
|
||||||
|
|||||||
Reference in New Issue
Block a user