diff --git a/pyproject.toml b/pyproject.toml
index a16fd81..f87bf87 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ dependencies = [
     "pybind11",
     "pylatexenc",
     "ray>=2.10",
-    "tensordict<0.6",
+    "tensordict<0.9",
     "torchdata",
     "transformers",
     "vllm<=0.6.3",
diff --git a/requirements.txt b/requirements.txt
index 8c35b11..c8eebf2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ pyarrow>=15.0.0
 pybind11
 pylatexenc
 ray[data,train,tune,serve]
-tensordict<0.6
+tensordict<0.9
 torchdata
 transformers
 vllm<=0.6.3