Skip to content

Commit ec3dbe3

Browse files
authored
Patch/dtype vectorizer from dict (#461)
Pass through dtype when creating from dict (do this in ret-opt)
1 parent 82776af commit ec3dbe3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

redisvl/utils/vectorize/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,12 @@
3737
def vectorizer_from_dict(
3838
vectorizer: dict,
3939
cache: dict = {},
40-
cache_folder=os.getenv("SENTENCE_TRANSFORMERS_HOME"),
4140
) -> BaseVectorizer:
4241
vectorizer_type = Vectorizers(vectorizer["type"])
4342
model = vectorizer["model"]
43+
dtype = vectorizer.get("dtype", "float32")
4444

45-
args = {"model": model}
45+
args = {"model": model, "dtype": dtype}
4646
if cache:
4747
emb_cache = EmbeddingsCache(**cache)
4848
args["cache"] = emb_cache

0 commit comments

Comments
 (0)