From a28a77bfe9f4de3ef1576da08499d820144b7d47 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Thu, 16 Nov 2023 15:20:08 +0100
Subject: [PATCH] put peft in requirements

---
 setup.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/setup.py b/setup.py
index faecb694d8..68cad46fbe 100644
--- a/setup.py
+++ b/setup.py
@@ -112,6 +112,7 @@ _deps = [
     "numpy",
     "omegaconf",
     "parameterized",
+    "peft==~0.6",
     "protobuf>=3.20.3,<4",
     "pytest",
     "pytest-timeout",
@@ -201,6 +202,7 @@ extras = {}
 extras["quality"] = deps_list("urllib3", "black", "isort", "ruff", "hf-doc-builder")
 extras["docs"] = deps_list("hf-doc-builder")
 extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2")
+extras["peft"] = deps_list("peft")
 extras["test"] = deps_list(
     "compel",
     "datasets",