diff --git a/requirements.txt b/requirements.txt index 667dd98..445189f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ pybind11 pylatexenc pylint==3.3.6 ray[default] -tensordict<=0.6.2 +tensordict<=0.10.0 torchdata transformers # vllm==0.6.3.post1 diff --git a/requirements_sglang.txt b/requirements_sglang.txt index e859e1c..6ef653a 100644 --- a/requirements_sglang.txt +++ b/requirements_sglang.txt @@ -12,7 +12,7 @@ pyarrow>=15.0.0 pybind11 pylatexenc ray[default]>=2.10 -tensordict<=0.6.2 +tensordict<=0.10.0 torchdata torchvision transformers