diff --git a/requirements-npu.txt b/requirements-npu.txt index fada1f8..1224ffd 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 1196072..4691c62 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 # vllm==0.8.4 diff --git a/requirements_sglang.txt b/requirements_sglang.txt index 113bca0..674526c 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