A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 14, 05:30 UTC
[ROCm] Add dependencies for ROCm (#24900)
Signed-off-by: Yida Wu <yida.wu@amd.com>
Concurrensee committed
Sep 16, 2025 at 14:49 UTC
dcf2f3ec067711ff69e5ab7478fca6ffb4f11daf
3 files changed
+4
-1
requirements/rocm-build.txt
+1
@@ -14,3 +14,4 @@ setuptools-scm>=8
14
wheel
15
jinja2>=3.1.6
16
amdsmi==6.2.4
17
+timm>=1.0.17
requirements/rocm-test.txt
+1
@@ -1,5 +1,6 @@
1
# Common dependencies
2
-r common.txt
3
+tblib==3.1.0
4
5
# entrypoints test
6
# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai
requirements/rocm.txt
+2
-1
@@ -17,4 +17,5 @@ setuptools>=77.0.3,<80.0.0
17
setuptools-scm>=8
18
runai-model-streamer==0.11.0
19
runai-model-streamer-s3==0.11.0
20
-conch-triton-kernels==1.2.1
\ No newline at end of file
20
+conch-triton-kernels==1.2.1
21
+timm>=1.0.17
\ No newline at end of file