Skip to content

Commit 36d901b

Browse files
Concurrenseecharlifu
authored andcommitted
[ROCm] Add dependencies for ROCm (vllm-project#24900)
Signed-off-by: Yida Wu <yida.wu@amd.com> Signed-off-by: charlifu <charlifu@amd.com>
1 parent 122b74e commit 36d901b

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

requirements/rocm-build.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,4 @@ setuptools-scm>=8
1414
wheel
1515
jinja2>=3.1.6
1616
amdsmi==6.2.4
17+
timm>=1.0.17

requirements/rocm-test.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# Common dependencies
22
-r common.txt
3+
tblib==3.1.0
34

45
# entrypoints test
56
# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai

requirements/rocm.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,5 @@ setuptools>=77.0.3,<80.0.0
1717
setuptools-scm>=8
1818
runai-model-streamer==0.11.0
1919
runai-model-streamer-s3==0.11.0
20-
conch-triton-kernels==1.2.1
20+
conch-triton-kernels==1.2.1
21+
timm>=1.0.17

0 commit comments

Comments
 (0)