Commit 09e91431 authored by Benjamin Fattori's avatar Benjamin Fattori
Browse files

accelerate added to requirements in setup.py

parent d4c5315a
...@@ -21,6 +21,7 @@ setuptools.setup( ...@@ -21,6 +21,7 @@ setuptools.setup(
], ],
python_requires=">=3.6", python_requires=">=3.6",
install_requires=[ install_requires=[
"accelerate>=0.18.0",
"datasets>=2.0.0", "datasets>=2.0.0",
"jsonlines", "jsonlines",
"numexpr", "numexpr",
......
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