Skip to content

Commit

Permalink
log
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Jul 15, 2024
1 parent 9ae5621 commit 5fe82b4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion py_txi/text_embedding_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ 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")
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:
Expand Down

0 comments on commit 5fe82b4

Please sign in to comment.