Commit 5797cabb authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Fix typo.

PiperOrigin-RevId: 434053895
parent 1d8fc5a6
...@@ -1234,7 +1234,7 @@ class Decoder(Module): ...@@ -1234,7 +1234,7 @@ class Decoder(Module):
encoded: the encoder outputs. encoded: the encoder outputs.
decoder_mask: the decoder self-attention mask. decoder_mask: the decoder self-attention mask.
encoder_decoder_mask: the cross-attention mask. encoder_decoder_mask: the cross-attention mask.
decode: Whether to perform autoaggressive decoding. decode: Whether to perform autoregressive decoding.
decode_position: integer, the position to decode. decode_position: integer, the position to decode.
cache: The cache dictionary of key, value tensors. cache: The cache dictionary of key, value tensors.
max_decode_len: An optional integer specifying the maximum decoding max_decode_len: An optional integer specifying the maximum decoding
......
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