Update modelling_hibial.py

modelling_hibial.py

@@ -333,9 +333,7 @@ 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 = torch.where(src_key_padding_mask, torch.tensor(float('-inf')), torch.tensor(0.0))
-        
+ 
         # memory = self.encoder(input_ids, mask=src_mask, src_key_padding_mask=src_key_padding_mask)
 
         # Encoding