Fix comment (#10886)

This commit is contained in:
lexhuismans 2021-03-25 19:23:56 +01:00 committed by GitHub
parent 9856c9213d
commit 86c6f8a8b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -904,6 +904,7 @@ class T5Stack(T5PreTrainedModel):
if past_key_values is None:
past_key_values = [None] * len(self.block)
# We can provide a self-attention mask of dimensions [batch_size, from_seq_length, to_seq_length]
# ourselves in which case we just need to make it broadcastable to all heads.
extended_attention_mask = self.get_extended_attention_mask(attention_mask, input_shape, inputs_embeds.device)