From dcf2f3ec067711ff69e5ab7478fca6ffb4f11daf Mon Sep 17 00:00:00 2001 From: Concurrensee Date: Tue, 16 Sep 2025 14:49:06 -0500 Subject: [PATCH] [ROCm] Add dependencies for ROCm (#24900) Signed-off-by: Yida Wu --- requirements/rocm-build.txt | 1 + requirements/rocm-test.txt | 1 + requirements/rocm.txt | 3 ++- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/requirements/rocm-build.txt b/requirements/rocm-build.txt index affe562c2..a86a8ab6d 100644 --- a/requirements/rocm-build.txt +++ b/requirements/rocm-build.txt @@ -14,3 +14,4 @@ setuptools-scm>=8 wheel jinja2>=3.1.6 amdsmi==6.2.4 +timm>=1.0.17 diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index 25f950a99..869fb28c3 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -1,5 +1,6 @@ # Common dependencies -r common.txt +tblib==3.1.0 # entrypoints test # librosa==0.10.2.post1 # required by audio tests in entrypoints/openai diff --git a/requirements/rocm.txt b/requirements/rocm.txt index 8e3995121..c129dd345 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -17,4 +17,5 @@ setuptools>=77.0.3,<80.0.0 setuptools-scm>=8 runai-model-streamer==0.11.0 runai-model-streamer-s3==0.11.0 -conch-triton-kernels==1.2.1 \ No newline at end of file +conch-triton-kernels==1.2.1 +timm>=1.0.17 \ No newline at end of file