chore: update from f36c08c8a58c21b5aaab523fa03fb4a24b475612

modeling_bert.py

@@ -1211,8 +1211,9 @@ class JinaBertModel(JinaBertPreTrainedModel):
             self.to(device)
 
         # TODO: Maybe use better length heuristic?
-        inverse_permutation = np.argsort(np.argsort([-len(i) for i in sentences]))
-        sentences = sorted(sentences, key=len, reverse=True)
+        permutation = np.argsort([-len(i) for i in sentences])
+        inverse_permutation = np.argsort(permutation)
+        sentences = [sentences[idx] for idx in permutation]
 
         padding = tokenizer_kwargs.pop('padding', True)