diff --git a/setup.py b/setup.py index 53ee988..546e9aa 100644 --- a/setup.py +++ b/setup.py @@ -22,6 +22,7 @@ "transformers", "DeepCache==0.1.1", "torch", + "accelerate", ], python_requires='>=3.8.0', )