diff --git a/pyproject.toml b/pyproject.toml
index 3a447e0..40e3ad1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ dependencies = [
     "pybind11",
     "pylatexenc",
     "ray>=2.10",
-    "tensordict<0.6",
+    "tensordict<0.8",
     "transformers",
     "vllm<=0.6.3",
     'wandb',
diff --git a/requirements.txt b/requirements.txt
index 8807b34..2bee51c 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.8
 transformers
 vllm<=0.6.3
 wandb