diff --git a/pyproject.toml b/pyproject.toml
index 427934f..a7756bf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,7 +40,7 @@ sentence-transformers = { version = "2.2.2", optional = true }
 
 datasets = { version = "2.13.1", optional = true }
 evaluate = { version = "0.4.0", optional = true }
-peft = { version = "0.4.0", optional = true }
+peft = { version = "0.5.0", optional = true }
 trl = { version = "0.4.7", optional = true }
 
 [tool.poetry.extras]
@@ -62,6 +62,7 @@ rag = [
     "tiktoken"
 ]
 rlhf = [
+    "scikit-learn",
     "transformers",
     "einops",
     "accelerate",