Skip to content

Commit 93ea26f

Browse files
DN6sayakpaul
andauthored
Add PEFT to training deps (#6148)
add peft to training deps Co-authored-by: Sayak Paul <[email protected]>
1 parent f5dfe2a commit 93ea26f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def run(self):
204204
extras = {}
205205
extras["quality"] = deps_list("urllib3", "isort", "ruff", "hf-doc-builder")
206206
extras["docs"] = deps_list("hf-doc-builder")
207-
extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2")
207+
extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2", "peft")
208208
extras["test"] = deps_list(
209209
"compel",
210210
"GitPython",

0 commit comments

Comments
 (0)