code adaptation/changes according to the commits on Oct 17, 2022

freewym · Oct 18, 2022 · 227ec47 · 227ec47
1 parent 060f04f
commit 227ec47
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/espresso/modules/sinusoidal_relative_positional_embedding.py b/espresso/modules/sinusoidal_relative_positional_embedding.py
@@ -87,12 +87,7 @@ def forward(
             relative posotional embedding for key of shape `(batch_size, 2*seq_len-1, embed_dim)`,
                 where `seq_len` is the length of key
         """
-        if torch.jit.is_scripting():
-            bspair = torch.onnx.operators.shape_as_tensor(input)
-        elif torch.onnx.is_in_onnx_export():
-            bspair = torch.onnx.operators.shape_as_tensor(input)
-        else:
-            bspair = input.size()
+        bspair = torch.onnx.operators.shape_as_tensor(input)
         bsz, seq_len = bspair[0], bspair[1]
         max_positions = self.weight.size(0)
         if self.padding_idx is not None: