diff --git a/inference/nlp/realtime/huggingface/sentence-transformers-triton-ensemble/examples/triton_sentence_embeddings.ipynb b/inference/nlp/realtime/huggingface/sentence-transformers-triton-ensemble/examples/triton_sentence_embeddings.ipynb index 5bd6ba0374..5fef5c4405 100644 --- a/inference/nlp/realtime/huggingface/sentence-transformers-triton-ensemble/examples/triton_sentence_embeddings.ipynb +++ b/inference/nlp/realtime/huggingface/sentence-transformers-triton-ensemble/examples/triton_sentence_embeddings.ipynb @@ -11,7 +11,7 @@ "source": [ "# Prerequisites\n", "\n", - "Install the necessary Python modules for using [NVIDIA Triton Inference Server](https://github.com/triton-inference-server/server/)." + "Install the necessary Python modules to use and interact with [NVIDIA Triton Inference Server](https://github.com/triton-inference-server/server/)." ] }, { @@ -25,7 +25,7 @@ }, "outputs": [], "source": [ - "! pip install sagemaker transformers==4.9.1 tritonclient[all]" + "! pip install torch==1.10.0 sagemaker transformers==4.9.1 tritonclient[all]" ] }, {