Update modelling_hibial.py

modelling_hibial.py

@@ -333,7 +333,8 @@ class HiBiAlBert(Module):
 
         bialibi_utt_mask[enc_mask_utt.bool()] = float('-inf')
         bialibi_ct_mask[enc_mask_ct.bool()] = float('-inf')
- 
+
+        src_key_padding_mask[src_key_padding_mask]=float('-inf')
         # memory = self.encoder(input_ids, mask=src_mask, src_key_padding_mask=src_key_padding_mask)
 
         # Encoding