diff --git a/backend/python/sglang/requirements-cpu.txt b/backend/python/sglang/requirements-cpu.txt index ad467ffa8..1e2dc6572 100644 --- a/backend/python/sglang/requirements-cpu.txt +++ b/backend/python/sglang/requirements-cpu.txt @@ -1,6 +1,6 @@ --extra-index-url https://download.pytorch.org/whl/cpu accelerate -torch==2.9.0 +torch==2.12.0+cpu torchvision torchaudio transformers diff --git a/backend/python/sglang/requirements-cublas12.txt b/backend/python/sglang/requirements-cublas12.txt index edd955820..749bc260e 100644 --- a/backend/python/sglang/requirements-cublas12.txt +++ b/backend/python/sglang/requirements-cublas12.txt @@ -6,7 +6,7 @@ # for cublas12 so uv consults this index alongside PyPI. --extra-index-url https://download.pytorch.org/whl/cu128 accelerate -torch==2.9.1 +torch==2.12.0+cpu torchvision torchaudio transformers diff --git a/backend/python/trl/requirements-cpu.txt b/backend/python/trl/requirements-cpu.txt index c67858542..56aa12cd4 100644 --- a/backend/python/trl/requirements-cpu.txt +++ b/backend/python/trl/requirements-cpu.txt @@ -1,9 +1,9 @@ --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.10.0 +torch==2.12.0+cpu trl peft datasets>=3.0.0 -transformers>=4.56.2 +transformers>=5.12.1 accelerate>=1.4.0 huggingface-hub>=1.3.0 sentencepiece diff --git a/backend/python/trl/requirements-cublas12.txt b/backend/python/trl/requirements-cublas12.txt index 05f29591c..843917ade 100644 --- a/backend/python/trl/requirements-cublas12.txt +++ b/backend/python/trl/requirements-cublas12.txt @@ -1,8 +1,8 @@ -torch==2.10.0 +torch==2.12.0+cpu trl peft datasets>=3.0.0 -transformers>=4.56.2 +transformers>=5.12.1 accelerate>=1.4.0 huggingface-hub>=1.3.0 sentencepiece diff --git a/backend/python/trl/requirements-cublas13.txt b/backend/python/trl/requirements-cublas13.txt index 05f29591c..843917ade 100644 --- a/backend/python/trl/requirements-cublas13.txt +++ b/backend/python/trl/requirements-cublas13.txt @@ -1,8 +1,8 @@ -torch==2.10.0 +torch==2.12.0+cpu trl peft datasets>=3.0.0 -transformers>=4.56.2 +transformers>=5.12.1 accelerate>=1.4.0 huggingface-hub>=1.3.0 sentencepiece diff --git a/backend/python/trl/requirements-mps.txt b/backend/python/trl/requirements-mps.txt index fbdfb6536..8f3f68841 100644 --- a/backend/python/trl/requirements-mps.txt +++ b/backend/python/trl/requirements-mps.txt @@ -1,8 +1,8 @@ -torch==2.10.0 +torch==2.12.0+cpu trl peft datasets>=3.0.0 -transformers>=4.56.2 +transformers>=5.12.1 accelerate>=1.4.0 huggingface-hub>=1.3.0 sentencepiece diff --git a/backend/python/vllm-omni/requirements-cublas12.txt b/backend/python/vllm-omni/requirements-cublas12.txt index deafb2c6e..5ce5539a5 100644 --- a/backend/python/vllm-omni/requirements-cublas12.txt +++ b/backend/python/vllm-omni/requirements-cublas12.txt @@ -1,4 +1,4 @@ accelerate -torch==2.7.0 +torch==2.12.0+cu130 transformers bitsandbytes