Merge pull request #67 from ethanyanjiali/update-cambioml-ethanli

Add scikit-learn and update PEFT to latest version
CambioML · Sep 20, 2023 · c6be98b · c6be98b
2 parents 3a9c511 + 09d3906
commit c6be98b
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",