Skip to content

Commit fcefbd5

Browse files
authored
pin torchvision deps if torch<2.4 (#2284)
1 parent 321e9b1 commit fcefbd5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

notebooks/distil-whisper-asr/distil-whisper-asr.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@
7070
},
7171
"outputs": [],
7272
"source": [
73-
"%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",
73+
"%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",
74+
"%pip install -q \"git+https://github.com/huggingface/optimum-intel.git\"\n",
7475
"%pip install -q \"openvino>=2023.2.0\" datasets \"gradio>=4.0\" \"librosa\" \"soundfile\"\n",
7576
"%pip install -q \"nncf>=2.6.0\" \"jiwer\""
7677
]

0 commit comments

Comments
 (0)