Installing torch to pass CI build

aws · Oct 5, 2022 · 94a2978 · 94a2978
1 parent 071f233
commit 94a2978
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ggingface/sentence-transformers-triton-ensemble/examples/triton_sentence_embeddings.ipynb b/...ggingface/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]"
    ]
   },
   {