search

huggingface / optimum-intel

🤗 Optimum Intel: Accelerate inference with Intel optimization tools
https://huggingface.co/docs/optimum/main/en/intel/index
Apache License 2.0
415 stars 112 forks source link

Install torchvision CPU in OpenVINO notebook tests #953

Closed helena-intel closed 1 month ago

helena-intel commented 1 month ago

Currently torch CPU version is installed, but torchvision CUDA version (as dependency of timm, which is a dependency of optimum-intel[tests]). This leads to issues. This tiny PR fixes that. Could also be fixed by using --extra-index-url everywhere, but this method is a bit safer and clearer.

HuggingFaceDocBuilderDev commented 1 month ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.