From 94a2978c6fd68cf82a0a074dd979ec217cc0523c Mon Sep 17 00:00:00 2001 From: joaopcm1996 Date: Wed, 5 Oct 2022 12:07:23 +0000 Subject: [PATCH] Installing torch to pass CI build --- .../examples/triton_sentence_embeddings.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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]" ] }, {