diff --git a/.ci/docker/requirements-ci.txt b/.ci/docker/requirements-ci.txt index f6ff1cf59f9..61bcb0ee9dd 100644 --- a/.ci/docker/requirements-ci.txt +++ b/.ci/docker/requirements-ci.txt @@ -3,6 +3,7 @@ numpy==1.25.2 PyYAML==6.0.1 ruamel.yaml==0.17.32 sympy==1.12 +timm==0.6.13 tomli==2.0.1 zstd==1.5.5.1 diff --git a/examples/install_requirements.sh b/examples/install_requirements.sh index 121167021d4..1ffe6425436 100755 --- a/examples/install_requirements.sh +++ b/examples/install_requirements.sh @@ -22,3 +22,6 @@ pip install --force-reinstall --pre torchvision=="${TORCH_VISION_VERSION}" -i ht TORCH_AUDIO_VERSION=2.1.0.dev20230813 pip install --force-reinstall --pre torchaudio=="${TORCH_AUDIO_VERSION}" -i https://download.pytorch.org/whl/nightly/cpu + +TIMM_VERSION=0.6.13 +pip install --pre timm=${TIMM_VERSION}