Raghavan commited on
Commit
e67c1c1
1 Parent(s): 66ed51c

Upload 7 files

Browse files
Files changed (1) hide show
  1. 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=encoder_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,