diff --git a/requirements-npu.txt b/requirements-npu.txt index ea197c98..4d0a040e 100644 --- a/requirements-npu.txt +++ b/requirements-npu.txt @@ -10,7 +10,7 @@ peft>=0.15.2 pyarrow>=15.0.0 pybind11 pylatexenc -tensordict>=0.8.0,<=0.10.0,!=0.9.0 +tensordict>=0.8.0,!=0.9.0,<=0.11.0 ray[default] wandb mathruler diff --git a/requirements.txt b/requirements.txt index 6a051b8e..96736ed0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ pybind11 pylatexenc pre-commit ray[default] -tensordict>=0.8.0,<=0.10.0,!=0.9.0 +tensordict>=0.8.0,!=0.9.0,<=0.11.0 torchdata transformers<5.0.0 # vllm==0.8.4 diff --git a/requirements_sglang.txt b/requirements_sglang.txt index 113bca0d..674526cc 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.8.0,<=0.10.0,!=0.9.0 +tensordict>=0.8.0,!=0.9.0,<=0.11.0 torchdata torchvision transformers