@tduncan / Nvidia-Optimization-Stack-VLLM Mirror · read-only / commits / dcf2f3ec06
Fork 0

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