Agents extras (#23301)

* Agents extras

* Add to docs
This commit is contained in:
Lysandre Debut
2023-05-11 20:25:51 +02:00
committed by GitHub
parent ab96bf0294
commit 71b19ee251
3 changed files with 27 additions and 3 deletions

View File

@@ -112,6 +112,7 @@ _deps = [
"datasets!=2.5.0",
"decord==0.6.0",
"deepspeed>=0.8.3",
"diffusers",
"dill<0.3.5",
"evaluate>=0.2.0",
"fairscale>0.3",
@@ -123,7 +124,7 @@ _deps = [
"fugashi>=1.0",
"GitPython<3.1.19",
"hf-doc-builder>=0.3.0",
"huggingface-hub>=0.11.0,<1.0",
"huggingface-hub>=0.14.1,<1.0",
"importlib_metadata",
"ipadic>=1.0.0,<2.0",
"isort>=5.5.4",
@@ -140,6 +141,7 @@ _deps = [
"onnxconverter-common",
"onnxruntime-tools>=1.4.2",
"onnxruntime>=1.4.0",
"opencv-python",
"optuna",
"optax>=0.0.8,<=0.1.4",
"packaging>=20.0",
@@ -412,6 +414,16 @@ extras["torchhub"] = deps_list(
"tqdm",
)
extras["agents"] = deps_list(
"diffusers",
"accelerate",
"datasets",
"torch",
"sentencepiece",
"opencv-python",
"Pillow"
)
# when modifying the following list, make sure to update src/transformers/dependency_versions_check.py
install_requires = [
deps["importlib_metadata"] + ";python_version<'3.8'", # importlib_metadata for Python versions that don't have it