update: import sentence-transformers

michaelfeil · Jun 13, 2024 · 26b9738 · 26b9738
1 parent 32f622d
commit 26b9738
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/infinity_emb/infinity_emb/transformer/quantization/interface.py b/libs/infinity_emb/infinity_emb/transformer/quantization/interface.py
@@ -5,7 +5,7 @@
 import numpy as np
 import requests  # type: ignore
 
-from infinity_emb._optional_imports import CHECK_TORCH
+from infinity_emb._optional_imports import CHECK_SENTENCE_TRANSFORMERS, CHECK_TORCH
 from infinity_emb.env import MANAGER
 from infinity_emb.log_handler import logger
 from infinity_emb.primitives import Device, Dtype, EmbeddingDtype
@@ -16,6 +16,8 @@
 
 if CHECK_TORCH.is_available:
     import torch
+
+if CHECK_SENTENCE_TRANSFORMERS.is_available:
     from sentence_transformers.quantization import quantize_embeddings  # type: ignore