Raghavan commited on
Commit
c65ba8d
1 Parent(s): 1094abe

Upload 7 files

Browse files
Files changed (1) hide show
  1. modeling_indictrans.py +1 -1
modeling_indictrans.py CHANGED
@@ -507,7 +507,7 @@ class IndicTransDecoderLayer(nn.Module):
507
  hidden_states, cross_attn_weights, cross_attn_present_key_value = self.encoder_attn(
508
  hidden_states=hidden_states,
509
  key_value_states=encoder_hidden_states,
510
- attention_mask=encoder_attention_mask,
511
  layer_head_mask=cross_attn_layer_head_mask,
512
  past_key_value=cross_attn_past_key_value,
513
  output_attentions=output_attentions,
 
507
  hidden_states, cross_attn_weights, cross_attn_present_key_value = self.encoder_attn(
508
  hidden_states=hidden_states,
509
  key_value_states=encoder_hidden_states,
510
+ attention_mask=None,
511
  layer_head_mask=cross_attn_layer_head_mask,
512
  past_key_value=cross_attn_past_key_value,
513
  output_attentions=output_attentions,