diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh
index 56e1f17..bc42627 100644
--- a/.github/unittest/install_dependencies.sh
+++ b/.github/unittest/install_dependencies.sh
@@ -4,8 +4,8 @@ python -m pip install --upgrade pip
 # Install required dependencies
 python -m pip install flake8 pytest pytest-cov hydra-core tqdm
 python -m pip install torch torchvision
-python -m pip install git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f
-python -m pip install git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd
+python -m pip install git+https://github.com/pytorch/rl.git@83a7a57daee4b76b249dbc0ce2231f163f2a1482
+python -m pip install git+https://github.com/pytorch/tensordict.git@c11024e0db864bd8487c68e3d0cf5e100fc07831
 
 # Install additional dependencies
 python -m pip install transformers promptsmiles MolScore
diff --git a/requirements.txt b/requirements.txt
index 75a1960..ccfba36 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,5 +8,5 @@ wandb
 rdkit>=2023.3.3
 
 # TorchRL and Tensordict
-git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f
-git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd
\ No newline at end of file
+git+https://github.com/pytorch/rl.git@83a7a57daee4b76b249dbc0ce2231f163f2a1482
+git+https://github.com/pytorch/tensordict.git@c11024e0db864bd8487c68e3d0cf5e100fc07831
\ No newline at end of file