diff --git a/PyTorch/built-in/rl/OpenRLHF_v0.5.7_for_PyTorch/requirements.txt b/PyTorch/built-in/rl/OpenRLHF_v0.5.7_for_PyTorch/requirements.txt index e6a68fc37556e324df7a3c490edd1059986d51c0..62650a0b2be712b9ed48883d8c59d14555f68080 100644 --- a/PyTorch/built-in/rl/OpenRLHF_v0.5.7_for_PyTorch/requirements.txt +++ b/PyTorch/built-in/rl/OpenRLHF_v0.5.7_for_PyTorch/requirements.txt @@ -12,6 +12,7 @@ optimum packaging peft pillow +pybind11==2.13.6 ray[default]==2.42.0 tensorboard torch==2.6.0 diff --git a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu-2_6.txt b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu-2_6.txt index 358c24a15a769d39a7ad0287c1e862eb5128908e..112daca50b6bdb2d12984de49819db95f5b0f67b 100644 --- a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu-2_6.txt +++ b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu-2_6.txt @@ -11,6 +11,7 @@ numpy==1.26.4 optimum packaging peft +pybind11==2.13.6 ray[default]==2.42.0 tensorboard torchmetrics diff --git a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu.txt b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu.txt index 46b524784f21341fafe724a387ae27e5d5678ea5..38a376a9bc457a548528cf24962f0b0c184d57bc 100644 --- a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu.txt +++ b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements-npu.txt @@ -9,6 +9,7 @@ loralib optimum packaging peft +pybind11==2.13.6 ray[default]==2.42.0 tensorboard torchmetrics diff --git a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements.txt b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements.txt index d31d8a74f097b591ac81e6e1fa6a49890f59a256..3418596bfc10bb3d9f61db86be10d42c1171490b 100644 --- a/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements.txt +++ b/PyTorch/built-in/rl/OpenRLHF_v0.6.2_for_PyTorch/requirements.txt @@ -11,6 +11,7 @@ optimum packaging peft pynvml>=12.0.0 +pybind11==2.13.6 ray[default]==2.42.0 tensorboard torch