Skip to content

Commit 4885608

Browse files
kirklandsignfacebook-github-bot
authored andcommitted
Add timm package to CI (#88)
Summary: This is required for some models like inception v4 Pull Request resolved: #88 Test Plan: After this is merged, we can test OSS with D48524577 Reviewed By: huydhn Differential Revision: D48568221 Pulled By: kirklandsign fbshipit-source-id: 09a8d27a99c3a501b84c668c0938a115f4225fb1
1 parent 0ece196 commit 4885608

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

.ci/docker/requirements-ci.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ numpy==1.25.2
33
PyYAML==6.0.1
44
ruamel.yaml==0.17.32
55
sympy==1.12
6+
timm==0.6.13
67
tomli==2.0.1
78
zstd==1.5.5.1
89

examples/install_requirements.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,3 +22,6 @@ pip install --force-reinstall --pre torchvision=="${TORCH_VISION_VERSION}" -i ht
2222

2323
TORCH_AUDIO_VERSION=2.1.0.dev20230813
2424
pip install --force-reinstall --pre torchaudio=="${TORCH_AUDIO_VERSION}" -i https://download.pytorch.org/whl/nightly/cpu
25+
26+
TIMM_VERSION=0.6.13
27+
pip install --pre timm=${TIMM_VERSION}

0 commit comments

Comments
 (0)