fix parallel worker init

Author: generall

fastembed/late_interaction/token_embeddings.py

@@ -1,4 +1,4 @@
-from typing import Union, Iterable, Optional, List, Dict, Any
+from typing import Union, Iterable, Optional, List, Dict, Any, Type
 
 import numpy as np
 
@@ -31,6 +31,10 @@ def list_supported_models(cls) -> List[Dict[str, Any]]:
         """
         return supported_token_embeddings_models
 
+    @classmethod
+    def _get_worker_class(cls) -> Type[TextEmbeddingWorker]:
+        return TokensEmbeddingWorker
+
     def _post_process_onnx_output(self, output: OnnxOutputContext) -> Iterable[np.ndarray]:
         # Size: (batch_size, sequence_length, hidden_size)
         embeddings = output.model_output