jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update HuggingFaceEmbeddings.py with default encode_kwargs and model_kwargs

Browse source
This commit is contained in:
anovazzi1 2024-01-16 11:58:15 -03:00
commit 7dcbe3a525
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/components/embeddings/HuggingFaceEmbeddings.py
View file

@ -21,8 +21,8 @@ class HuggingFaceEmbeddingsComponent(CustomComponent):
def build( def build(
self, self,
cache_folder: Optional[str] = None, cache_folder: Optional[str] = None,
encode_kwargs: Optional[Dict] = None, encode_kwargs: Optional[Dict] = {},
model_kwargs: Optional[Dict] = None, model_kwargs: Optional[Dict] = {},
model_name: str = "sentence-transformers/all-mpnet-base-v2", model_name: str = "sentence-transformers/all-mpnet-base-v2",
multi_process: bool = False, multi_process: bool = False,
) -> HuggingFaceEmbeddings: ) -> HuggingFaceEmbeddings: