diff --git a/requirements-npu.txt b/requirements-npu.txt index 579b76a9..b4c331ce 100644 --- a/requirements-npu.txt +++ b/requirements-npu.txt @@ -11,7 +11,7 @@ peft>=0.15.2 pyarrow>=15.0.0 pybind11 pylatexenc -tensordict>=0.8.0,<=0.10.0,!=0.9.0 +tensordict!=0.9.0,<=0.13.0,>=0.13.0 ray[default] wandb mathruler diff --git a/requirements.txt b/requirements.txt index c39ff989..3f71e364 100644 --- a/requirements.txt +++ b/requirements.txt @@ -26,7 +26,7 @@ pybind11 pylatexenc pre-commit ray[default]>=2.41.0 -tensordict>=0.8.0,<=0.10.0,!=0.9.0 +tensordict!=0.9.0,<=0.13.0,>=0.13.0 torchdata transformers wandb