Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "91b8478c84af1f0a5de1f489487714cd6cead329"
Unverified Commit e72f1a8a authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Add torchvision to training deps (#607)

parent 4f1c989f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ e72f1a8a
...@@ -174,7 +174,7 @@ extras = {} ...@@ -174,7 +174,7 @@ extras = {}
extras["quality"] = ["black==22.8", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"] extras["quality"] = ["black==22.8", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"]
extras["docs"] = ["hf-doc-builder"] extras["docs"] = ["hf-doc-builder"]
extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"] extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"]
extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "transformers"] extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "torchvision", "transformers"]
extras["torch"] = deps_list("torch") extras["torch"] = deps_list("torch")
if os.name == "nt": # windows if os.name == "nt": # windows
......
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