lgcharpe commited on
Commit
93b66fd
1 Parent(s): d06c1d3

Update modeling_norbert.py

Browse files
Files changed (1) hide show
  1. modeling_norbert.py +2 -2
modeling_norbert.py CHANGED
@@ -162,8 +162,8 @@ class Attention(nn.Module):
162
  if self.position_indices.size(0) < query_len:
163
  position_indices = torch.arange(query_len, dtype=torch.long).unsqueeze(1) \
164
  - torch.arange(query_len, dtype=torch.long).unsqueeze(0)
165
- position_indices = self.make_log_bucket_position(position_indices, self.position_bucket_size, 512)
166
- position_indices = self.position_bucket_size - 1 + position_indices
167
  self.position_indices = position_indices.to(hidden_states.device)
168
 
169
  hidden_states = self.pre_layer_norm(hidden_states)
 
162
  if self.position_indices.size(0) < query_len:
163
  position_indices = torch.arange(query_len, dtype=torch.long).unsqueeze(1) \
164
  - torch.arange(query_len, dtype=torch.long).unsqueeze(0)
165
+ position_indices = self.make_log_bucket_position(position_indices, self.config.position_bucket_size, 512)
166
+ position_indices = self.config.position_bucket_size - 1 + position_indices
167
  self.position_indices = position_indices.to(hidden_states.device)
168
 
169
  hidden_states = self.pre_layer_norm(hidden_states)