Fix extrapolation in relative transformer, fix: #1906

Author: hankcs

hanlp/layers/transformers/relative_transformer.py

@@ -70,7 +70,7 @@ def forward(self, inputs: Tensor):
         """
         bsz, seq_len = inputs.size()
         max_pos = self.padding_idx + seq_len
-        if max_pos > self.origin_shift:
+        if max_pos >= self.origin_shift:
             # recompute/expand embeddings if needed
             weights = self.get_embedding(
                 max_pos * 2,

hanlp/version.py

@@ -2,7 +2,7 @@
 # Author: hankcs
 # Date: 2019-12-28 19:26
 
-__version__ = '2.1.0-beta.60'
+__version__ = '2.1.0-beta.61'
 """HanLP version"""