From 989acbf1c87d91ea3f80608668ada787df667f65 Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Thu, 14 Mar 2024 11:23:45 -0600 Subject: [PATCH] Add explicit PyTorch install for LLM Converter We are removing this from the explicit dependencies due to slow install time. --- examples/llm_inference/conversion/llm_conversion.ipynb | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/llm_inference/conversion/llm_conversion.ipynb b/examples/llm_inference/conversion/llm_conversion.ipynb index d75bc6ee..21306c6b 100644 --- a/examples/llm_inference/conversion/llm_conversion.ipynb +++ b/examples/llm_inference/conversion/llm_conversion.ipynb @@ -740,6 +740,7 @@ "display(install_out)\n", "with install_out:\n", " !pip install mediapipe\n", + " !pip install torch\n", " !pip install huggingface_hub\n", " import os\n", " from huggingface_hub import hf_hub_download\n",