Skip to content

Commit a28a77b

Browse files
put peft in requirements
1 parent fbb8b34 commit a28a77b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@
112112
"numpy",
113113
"omegaconf",
114114
"parameterized",
115+
"peft==~0.6",
115116
"protobuf>=3.20.3,<4",
116117
"pytest",
117118
"pytest-timeout",
@@ -201,6 +202,7 @@ def run(self):
201202
extras["quality"] = deps_list("urllib3", "black", "isort", "ruff", "hf-doc-builder")
202203
extras["docs"] = deps_list("hf-doc-builder")
203204
extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2")
205+
extras["peft"] = deps_list("peft")
204206
extras["test"] = deps_list(
205207
"compel",
206208
"datasets",

0 commit comments

Comments
 (0)