diff --git a/requirements.txt b/requirements.txt
index 1ce8baf..9955bf8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,7 +15,7 @@ pybind11
 pylatexenc
 pre-commit
 ray[default]
-tensordict<=0.6.2
+tensordict<=0.8.1
 torchdata
 transformers
 # vllm==0.6.3.post1
diff --git a/requirements_sglang.txt b/requirements_sglang.txt
index ac43c65..844f797 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.8.1
 torchdata
 torchvision
 transformers