From 86b35105c1ff4fab2c8f165744c49cc078dbfbf7 Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Sat, 27 May 2023 21:49:42 -0700
Subject: [PATCH] deps: add beginning requirements for fine-tune

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 pyproject.toml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 4ec41790..af943a4a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -67,6 +67,9 @@ dependencies = [
     "black[jupyter]==23.3.0",
 ]
 
+[project.optional-dependencies]
+fine-tune = ["peft", "bitsandbytes"]
+
 [project.urls]
 Documentation = "https://github.com/llmsys/openllm#readme"
 Issues = "https://github.com/llmsys/openllm/issues"