diff --git a/t5x/models.py b/t5x/models.py index 89e3784e9..4ca07837c 100644 --- a/t5x/models.py +++ b/t5x/models.py @@ -602,7 +602,7 @@ def predict_batch_with_aux( This method can be used with a customizable decoding function as long as it follows the signature of `DecodeFnCallable`. In order to provide a unified - interface for the decoding functions, we use a generic names. For example, a + interface for the decoding functions, we use generic names. For example, a beam size is a concept unique to beam search. Conceptually, it corresponds to the number of sequences returned by the beam search. Therefore, the generic argument `num_decodes` corresponds to the beam size if