Upload 7 files
Browse files- modeling_indictrans.py +1 -1
modeling_indictrans.py
CHANGED
@@ -505,7 +505,7 @@ class IndicTransDecoderLayer(nn.Module):
|
|
505 |
hidden_states, cross_attn_weights, cross_attn_present_key_value = self.encoder_attn(
|
506 |
hidden_states=hidden_states,
|
507 |
key_value_states=encoder_hidden_states,
|
508 |
-
attention_mask=
|
509 |
layer_head_mask=cross_attn_layer_head_mask,
|
510 |
past_key_value=cross_attn_past_key_value,
|
511 |
output_attentions=output_attentions,
|
|
|
505 |
hidden_states, cross_attn_weights, cross_attn_present_key_value = self.encoder_attn(
|
506 |
hidden_states=hidden_states,
|
507 |
key_value_states=encoder_hidden_states,
|
508 |
+
attention_mask=None,
|
509 |
layer_head_mask=cross_attn_layer_head_mask,
|
510 |
past_key_value=cross_attn_past_key_value,
|
511 |
output_attentions=output_attentions,
|