Add PEFT to training deps (#6148)

add peft to training deps Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

Add PEFT to training deps (#6148)
add peft to training deps Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
93ea26f2 · Dhruv Nair · GitHub · f5dfe2a8 · 93ea26f2
Unverified Commit 93ea26f2 authored Dec 15, 2023 by Dhruv Nair Committed by GitHub Dec 15, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -204,7 +204,7 @@ class DepsTableUpdateCommand(Command):
 extras = {}
 extras["quality"] = deps_list("urllib3", "isort", "ruff", "hf-doc-builder")
 extras["docs"] = deps_list("hf-doc-builder")
-extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2")
+extras["training"] = deps_list("accelerate", "datasets", "protobuf", "tensorboard", "Jinja2", "peft")
 extras["test"] = deps_list(
    "compel",
    "GitPython",