Update modeling_norbert.py
Browse files- 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)
|