@@ -71,6 +71,7 @@ __init__(
7171 local_files_only: bool = False ,
7272 meta_fields_to_embed: list[str ] | None = None ,
7373 embedding_separator: str = " \n " ,
74+ model_kwargs: dict[str , Any] | None = None ,
7475) -> None
7576```
7677
@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495- ** local_files_only** (<code >bool</code >) – If ` True ` , only use the model files in the ` cache_dir ` .
9596- ** meta_fields_to_embed** (<code >list\[ str\] | None</code >) – List of meta fields that should be embedded along with the Document content.
9697- ** embedding_separator** (<code >str</code >) – Separator used to concatenate the meta fields to the Document content.
98+ - ** model_kwargs** (<code >dict\[ str, Any\] | None</code >) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+ such as ` providers ` (e.g. ` ["CUDAExecutionProvider"] ` to run on GPU), ` cuda ` , or ` device_ids ` .
97100
98101#### to_dict
99102
@@ -198,7 +201,7 @@ __init__(
198201) -> None
199202```
200203
201- Create an FastembedDocumentEmbedder component.
204+ Create a FastembedSparseDocumentEmbedder component.
202205
203206** Parameters:**
204207
@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349** Returns:**
347350
348351- <code >dict\[ str, SparseEmbedding\] </code > – A dictionary with the following keys:
349- - ` embedding ` : A list of floats representing the embedding of the input text.
352+ - ` sparse_embedding ` : The ` SparseEmbedding ` of the input text.
350353
351354** Raises:**
352355
@@ -385,6 +388,7 @@ __init__(
385388 progress_bar: bool = True ,
386389 parallel: int | None = None ,
387390 local_files_only: bool = False ,
391+ model_kwargs: dict[str , Any] | None = None ,
388392) -> None
389393```
390394
@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408 If 0, use all available cores.
405409 If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410- ** local_files_only** (<code >bool</code >) – If ` True ` , only use the model files in the ` cache_dir ` .
411+ - ** model_kwargs** (<code >dict\[ str, Any\] | None</code >) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+ such as ` providers ` (e.g. ` ["CUDAExecutionProvider"] ` to run on GPU), ` cuda ` , or ` device_ids ` .
407413
408414#### to_dict
409415
0 commit comments