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(
self,
cache_folder: Optional[str] = None,
encode_kwargs: Optional[Dict] = None,
model_kwargs: Optional[Dict] = None,
encode_kwargs: Optional[Dict] = {},
model_kwargs: Optional[Dict] = {},
model_name: str = "sentence-transformers/all-mpnet-base-v2",
multi_process: bool = False,
) -> HuggingFaceEmbeddings: