add accelerate dependency

Summary: Accelerate is an additional implicitron dependency, so document it.

Reviewed By: shapovalov

Differential Revision: D37786933

fbshipit-source-id: 11024fe604107881f8ca29e17cb5cbfe492fc7f9
This commit is contained in:
Jeremy Reizenstein
2022-07-13 06:00:05 -07:00
committed by Facebook GitHub Bot
parent 4261e59f51
commit 55f67b0d18
2 changed files with 11 additions and 4 deletions

View File

@@ -143,8 +143,15 @@ setup(
install_requires=["fvcore", "iopath"],
extras_require={
"all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"],
"dev": ["flake8", "usort", "black==22.3"],
"implicitron": ["hydra-core>=1.1", "visdom", "lpips", "matplotlib"],
"dev": ["flake8", "usort"],
"implicitron": [
"hydra-core>=1.1",
"visdom",
"lpips",
"tqdm>4.29.0",
"matplotlib",
"accelerate",
],
},
entry_points={
"console_scripts": [