Skip to content

GitLab

Commit c77f17b9 authored by zhuwenwen's avatar zhuwenwen
Browse files

update deps

parent 55f7b089
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 3 deletions
requirements/rocm.txt
View file @ c77f17b9
...@@ -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,12 +20,15 @@ runai-model-streamer==0.11.0 ...@@ -19,12 +20,15 @@ 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.0.0 triton == 3.1
flash_attn == 2.6.1 flash_attn == 2.6.1
flash_mla == 1.0.0 flash_mla == 1.0.0
lightop == 0.5.0 lightop == 0.6.0
lmslim == 0.3.1 lmslim == 0.3.1
numa numa
python-multipart python-multipart
......
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