wip

CambioML · Sep 20, 2023 · 09d3906 · 09d3906
1 parent 3a9c511
commit 09d3906
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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",