Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "53d99671bd150ca47f5710822ddde9589816c163"
Unverified Commit 93ea26f2 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Add PEFT to training deps (#6148) 



add peft to training deps
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent f5dfe2a8
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 93ea26f2
...@@ -204,7 +204,7 @@ class DepsTableUpdateCommand(Command): ...@@ -204,7 +204,7 @@ class DepsTableUpdateCommand(Command):
extras = {} extras = {}
extras["quality"] = deps_list("urllib3", "isort", "ruff", "hf-doc-builder") extras["quality"] = deps_list("urllib3", "isort", "ruff", "hf-doc-builder")
extras["docs"] = deps_list("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( extras["test"] = deps_list(
"compel", "compel",
"GitPython", "GitPython",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment