diff --git a/py_txi/text_embedding_inference.py b/py_txi/text_embedding_inference.py index f817c16..640adb9 100644 --- a/py_txi/text_embedding_inference.py +++ b/py_txi/text_embedding_inference.py @@ -31,8 +31,8 @@ def __post_init__(self) -> None: LOGGER.info("\t+ Using the latest NVIDIA GPU image for Text-Embedding-Inference") 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" + LOGGER.info("\t+ Using version 1.4 image for Text-Embedding-Inference") + 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.")