Skip to content

GitLab

Unverified Commit dcf2f3ec authored by Concurrensee's avatar Concurrensee Committed by GitHub
Browse files

[ROCm] Add dependencies for ROCm (#24900) 


Signed-off-by: default avatarYida Wu <yida.wu@amd.com>
parent 218454b9
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
requirements/rocm-build.txt
View file @ dcf2f3ec
...@@ -14,3 +14,4 @@ setuptools-scm>=8 ...@@ -14,3 +14,4 @@ setuptools-scm>=8
wheel wheel
jinja2>=3.1.6 jinja2>=3.1.6
amdsmi==6.2.4 amdsmi==6.2.4
timm>=1.0.17
requirements/rocm-test.txt
View file @ dcf2f3ec
# Common dependencies # Common dependencies
-r common.txt -r common.txt
tblib==3.1.0
# entrypoints test # entrypoints test
# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai # librosa==0.10.2.post1 # required by audio tests in entrypoints/openai
......
requirements/rocm.txt
View file @ dcf2f3ec
...@@ -18,3 +18,4 @@ setuptools-scm>=8 ...@@ -18,3 +18,4 @@ setuptools-scm>=8
runai-model-streamer==0.11.0 runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0 runai-model-streamer-s3==0.11.0
conch-triton-kernels==1.2.1 conch-triton-kernels==1.2.1
timm>=1.0.17
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment