Skip to content

Commit 297effb

Browse files
committed
test: move timm to test requirements
1 parent 488399c commit 297effb

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.github/workflows/unit-test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ jobs:
5959
python3 -m pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118
6060
- name: Install lmdeploy
6161
run: |
62-
python3 -m pip install pynvml packaging protobuf transformers_stream_generator matplotlib timm
62+
python3 -m pip install pynvml packaging protobuf transformers_stream_generator matplotlib
6363
# manually install flash attn
6464
python3 -m pip install /root/packages/cu118/flash_attn-*.whl
6565
python3 -m pip install -r requirements_cuda.txt -r requirements/test.txt

requirements/test.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,4 @@ pytest-rerunfailures
1010
pytest-sugar
1111
pytest-xdist
1212
pyyaml
13+
timm

0 commit comments

Comments
 (0)