We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 82776af commit ec3dbe3Copy full SHA for ec3dbe3
redisvl/utils/vectorize/__init__.py
@@ -37,12 +37,12 @@
37
def vectorizer_from_dict(
38
vectorizer: dict,
39
cache: dict = {},
40
- cache_folder=os.getenv("SENTENCE_TRANSFORMERS_HOME"),
41
) -> BaseVectorizer:
42
vectorizer_type = Vectorizers(vectorizer["type"])
43
model = vectorizer["model"]
+ dtype = vectorizer.get("dtype", "float32")
44
45
- args = {"model": model}
+ args = {"model": model, "dtype": dtype}
46
if cache:
47
emb_cache = EmbeddingsCache(**cache)
48
args["cache"] = emb_cache
0 commit comments