Embeddings are sent back to the CPU, so they can be converted to nump…

…y arrays

src/seb/registered_models/e5_mistral.py

@@ -183,7 +183,7 @@ def encode(
                 )
             batched_embeddings.append(embeddings)
 
-        return torch.cat(batched_embeddings)
+        return torch.cat(batched_embeddings).to("cpu")
 
     def encode_corpus(self, corpus: list[dict[str, str]], **kwargs: Any):
         if isinstance(corpus, dict):