diff --git a/requirements-npu.txt b/requirements-npu.txt index 601e8f9f..60b4849b 100644 --- a/requirements-npu.txt +++ b/requirements-npu.txt @@ -11,7 +11,7 @@ pyarrow>=15.0.0 pybind11 pylatexenc ray -tensordict<=0.6.2 +tensordict<=0.8.3 transformers>=4.52.0 wandb mathruler diff --git a/requirements.txt b/requirements.txt index 3d2d80e2..afe6eadb 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ pybind11 pylatexenc pre-commit ray[default] -tensordict<=0.6.2 +tensordict<=0.8.3 torchdata transformers==4.51.1 # vllm==0.8.4 diff --git a/requirements_sglang.txt b/requirements_sglang.txt index 57d5e0be..bbbf2c86 100644 --- a/requirements_sglang.txt +++ b/requirements_sglang.txt @@ -12,7 +12,7 @@ pyarrow>=19.0.0 pybind11 pylatexenc ray[default]>=2.10 -tensordict<=0.6.2 +tensordict<=0.8.3 torchdata torchvision transformers