pin torchvision deps if torch<2.4 (#2284)

openvinotoolkit · Aug 12, 2024 · fcefbd5 · fcefbd5
1 parent 321e9b1
commit fcefbd5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/distil-whisper-asr/distil-whisper-asr.ipynb b/notebooks/distil-whisper-asr/distil-whisper-asr.ipynb
@@ -70,7 +70,8 @@
    },
    "outputs": [],
    "source": [
-    "%pip install -q \"transformers>=4.35\" \"torch>=2.1,<2.4.0\" onnx \"git+https://github.com/huggingface/optimum-intel.git\" \"peft==0.6.2\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
+    "%pip install -q \"transformers>=4.35\" \"torch>=2.1,<2.4.0\" \"torchvision<0.19.0\" onnx \"peft==0.6.2\" --extra-index-url https://download.pytorch.org/whl/cpu\n",
+    "%pip install -q \"git+https://github.com/huggingface/optimum-intel.git\"\n",
     "%pip install -q \"openvino>=2023.2.0\" datasets  \"gradio>=4.0\" \"librosa\" \"soundfile\"\n",
     "%pip install -q \"nncf>=2.6.0\" \"jiwer\""
    ]