Compare commits

...

2 Commits

Author SHA1 Message Date
Sayak Paul
b08fc2d9d5 Merge branch 'main' into examples-test-fix 2023-12-14 21:49:52 +05:30
Dhruv Nair
a031abdc89 add peft to training deps 2023-12-12 14:07:45 +00:00

View File

@@ -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",