pin TEI cpu version

IlyasMoutawwakil · Jul 15, 2024 · 9ae5621 · 9ae5621
1 parent 90cd353
commit 9ae5621
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py_txi/text_embedding_inference.py b/py_txi/text_embedding_inference.py
@@ -32,7 +32,7 @@ def __post_init__(self) -> None:
                 self.image = "ghcr.io/huggingface/text-embeddings-inference:latest"
             else:
                 LOGGER.info("\t+ Using the latest CPU image for Text-Embedding-Inference")
-                self.image = "ghcr.io/huggingface/text-embeddings-inference:cpu-latest"
+                self.image = "ghcr.io/huggingface/text-embeddings-inference:cpu-1.4"
 
         if is_nvidia_system() and "cpu" in self.image:
             LOGGER.warning("\t+ You are running on a NVIDIA GPU system but using a CPU image.")