From 09d3906cd4b7af4c7c223f8ac71a61cf51e652d9 Mon Sep 17 00:00:00 2001
From: Ethan Yanjia Li <liyanjia92@gmail.com>
Date: Tue, 19 Sep 2023 23:14:39 -0700
Subject: [PATCH] wip

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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",