diff --git a/.ci/docker/common/install_inductor_benchmark_deps.sh b/.ci/docker/common/install_inductor_benchmark_deps.sh index 7a0fefe8850c4..5b775af6539ff 100644 --- a/.ci/docker/common/install_inductor_benchmark_deps.sh +++ b/.ci/docker/common/install_inductor_benchmark_deps.sh @@ -14,13 +14,6 @@ function install_timm() { local commit commit=$(get_pinned_commit timm) - # TODO (huydhn): There is no torchvision release on 3.13 when I write this, so - # I'm using nightly here instead. We just need to package to be able to install - # TIMM. Removing this once vision has a release on 3.13 - if [[ "${ANACONDA_PYTHON_VERSION}" == "3.13" ]]; then - pip_install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 - fi - pip_install "git+https://github.com/huggingface/pytorch-image-models@${commit}" # Clean up conda_run pip uninstall -y cmake torch torchvision triton