Skip to content

Commit

Permalink
Allowing HuggingFaceEmbeddings from the cached weight (langchain-ai#3084
Browse files Browse the repository at this point in the history
)

### langchain-ai#3079
Allow initializing HuggingFaceEmbeddings from the cached weight
  • Loading branch information
azamiftikhar1000 authored and samching committed May 1, 2023
1 parent 53628e4 commit 8f481a1
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions langchain/embeddings/huggingface.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
"""Wrapper around HuggingFace embedding models."""
from typing import Any, List
from typing import Any, List, Optional

from pydantic import BaseModel, Extra

Expand Down Expand Up @@ -29,14 +29,19 @@ class HuggingFaceEmbeddings(BaseModel, Embeddings):
client: Any #: :meta private:
model_name: str = DEFAULT_MODEL_NAME
"""Model name to use."""
cache_folder: Optional[str] = None
"""Path to store models.
Can be also set by SENTENCE_TRANSFORMERS_HOME enviroment variable."""

def __init__(self, **kwargs: Any):
"""Initialize the sentence_transformer."""
super().__init__(**kwargs)
try:
import sentence_transformers

self.client = sentence_transformers.SentenceTransformer(self.model_name)
self.client = sentence_transformers.SentenceTransformer(
self.model_name, self.cache_folder
)
except ImportError:
raise ValueError(
"Could not import sentence_transformers python package. "
Expand Down

0 comments on commit 8f481a1

Please sign in to comment.