Skip to content

Commit 8baece8

Browse files
committed
drop use of cache folder
1 parent 488b692 commit 8baece8

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

test/unit/v2/embedders/test_huggingface.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ def generate_embedder_config_params() -> dict:
2727
if random.random() < 0.5
2828
else None
2929
)
30-
params["cache_folder"] = fake.file_path() if random.random() < 0.5 else None
3130
return params
3231

3332

unstructured_ingest/embed/huggingface.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ class HuggingFaceEmbeddingConfig(EmbeddingConfig):
2222
default_factory=lambda: {"device": "cpu"}, alias="model_kwargs"
2323
)
2424
encode_kwargs: Optional[dict] = Field(default_factory=lambda: {"normalize_embeddings": False})
25-
cache_folder: Optional[str] = Field(default=None)
2625

2726
@requires_dependencies(
2827
["sentence_transformers"],
@@ -33,7 +32,6 @@ def get_client(self) -> "SentenceTransformer":
3332

3433
return SentenceTransformer(
3534
model_name_or_path=self.embedder_model_name,
36-
cache_folder=self.cache_folder,
3735
**self.embedder_model_kwargs,
3836
)
3937

0 commit comments

Comments
 (0)