We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 488399c commit 297effbCopy full SHA for 297effb
.github/workflows/unit-test.yml
@@ -59,7 +59,7 @@ jobs:
59
python3 -m pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118
60
- name: Install lmdeploy
61
run: |
62
- python3 -m pip install pynvml packaging protobuf transformers_stream_generator matplotlib timm
+ python3 -m pip install pynvml packaging protobuf transformers_stream_generator matplotlib
63
# manually install flash attn
64
python3 -m pip install /root/packages/cu118/flash_attn-*.whl
65
python3 -m pip install -r requirements_cuda.txt -r requirements/test.txt
requirements/test.txt
@@ -10,3 +10,4 @@ pytest-rerunfailures
10
pytest-sugar
11
pytest-xdist
12
pyyaml
13
+timm
0 commit comments