Install torchvision CPU in OpenVINO notebook tests (#953)

huggingface · Oct 17, 2024 · f7b5b54 · f7b5b54
1 parent 981442f
commit f7b5b54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/test_openvino_notebooks.yml b/.github/workflows/test_openvino_notebooks.yml
@@ -39,7 +39,7 @@ jobs:
         # Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages
         # ffmpeg, torchaudio and pillow are required for image classification and audio classification pipelines
         sudo apt-get install ffmpeg
-        pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
         pip install -r notebooks/openvino/requirements.txt
         pip install .[tests,openvino] nbval