diff --git a/requirements.txt b/requirements.txt
index 1ce8baf..2fe4272 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,7 +15,7 @@ pybind11
 pylatexenc
 pre-commit
 ray[default]
-tensordict<=0.6.2
+tensordict<=0.9.0
 torchdata
 transformers
 # vllm==0.6.3.post1
diff --git a/requirements_sglang.txt b/requirements_sglang.txt
index ac43c65..3149409 100644
--- a/requirements_sglang.txt
+++ b/requirements_sglang.txt
@@ -13,7 +13,7 @@ pyarrow>=15.0.0
 pybind11
 pylatexenc
 ray[default]>=2.10
-tensordict<=0.6.2
+tensordict<=0.9.0
 torchdata
 torchvision
 transformers