From 66839cf36a431449e2a7389c27b039dffabb65a4 Mon Sep 17 00:00:00 2001 From: "d.rudenko" Date: Wed, 15 Jan 2025 12:23:24 +0100 Subject: [PATCH] Fix --- fastembed/text/onnx_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastembed/text/onnx_embedding.py b/fastembed/text/onnx_embedding.py index ff53582a..07ac6882 100644 --- a/fastembed/text/onnx_embedding.py +++ b/fastembed/text/onnx_embedding.py @@ -185,7 +185,7 @@ class OnnxTextEmbedding(TextEmbeddingBase, OnnxTextModel[np.ndarray]): """Implementation of the Flag Embedding model.""" @classmethod - def list_supported_models(cls) -> List[Dict[str, Any]]: + def list_supported_models(cls) -> list[Dict[str, Any]]: """ Lists the supported models. @@ -217,7 +217,7 @@ def __init__( Mutually exclusive with the `cuda` and `device_ids` arguments. Defaults to None. cuda (bool, optional): Whether to use cuda for inference. Mutually exclusive with `providers` Defaults to False. - device_ids (Optional[List[int]], optional): The list of device ids to use for data parallel processing in + device_ids (Optional[list[int]], optional): The list of device ids to use for data parallel processing in workers. Should be used with `cuda=True`, mutually exclusive with `providers`. Defaults to None. lazy_load (bool, optional): Whether to load the model during class initialization or on demand. Should be set to True when using multiple-gpu and parallel encoding. Defaults to False.