Skip to content

GitLab

Commit 97f936be authored by Benjamin Fattori's avatar Benjamin Fattori
Browse files

add accelerate to deps

parent f1dedbf0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
setup.py
View file @ 97f936be
...@@ -37,6 +37,7 @@ setuptools.setup( ...@@ -37,6 +37,7 @@ setuptools.setup(
"tqdm-multiprocess", "tqdm-multiprocess",
"transformers>=4.1", "transformers>=4.1",
"zstandard", "zstandard",
"accelerate>=0.17.1"
], ],
extras_require={ extras_require={
"dev": ["black", "flake8", "pre-commit", "pytest", "pytest-cov"], "dev": ["black", "flake8", "pre-commit", "pytest", "pytest-cov"],
......
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