Skip to content

GitLab

Commit 14af6a70 authored by zhuwenwen's avatar zhuwenwen
Browse files

add deps of pd-pp

parent 02761ab1
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
requirements/rocm.txt
View file @ 14af6a70
...@@ -8,7 +8,8 @@ numba == 0.61.2; python_version > '3.9' ...@@ -8,7 +8,8 @@ numba == 0.61.2; python_version > '3.9'
boto3 boto3
botocore botocore
datasets datasets
ray>=2.10.0,<2.45.0 # ray>=2.10.0,<2.45.0
ray==2.48.0
peft peft
pytest-asyncio pytest-asyncio
tensorizer>=2.9.0 tensorizer>=2.9.0
...@@ -19,6 +20,9 @@ runai-model-streamer==0.11.0 ...@@ -19,6 +20,9 @@ runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0 runai-model-streamer-s3==0.11.0
setuptools_scm>=8 setuptools_scm>=8
cmake==3.29 cmake==3.29
quart
fastrlock==0.8.3
cupy==12.3.0
torch == 2.5.1 torch == 2.5.1
triton == 3.1 triton == 3.1
......
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